qimpy.pre_init

Initialization that must occur before all other imports: - Create log. - Set GPU visibility before importing MPI and torch when possible.

Module Attributes

log	Log for the qimpy module, configurable using qimpy.mpi.log_config()

Functions

set_gpu_visibility

Update CUDA_VISIBLE_DEVICES to select one GPU based on local_rank of process.