NVIDIA Environment Variable Configuration
· ☕ 2 min read
NVIDIA_VISIBLE_DEVICES Specifies the GPU devices visible to the program
1 CUDA_VISIBLE_DEVICES=0,1 Available values:
1,2, a comma-separated list of GPU UUIDs or indices all, all GPUs none, load the driver but with no GPU access void, do not load the driver NVIDIA_DRIVER_CAPABILITIES Controls which driver libraries/binaries will be installed inside the container