[nvidia] CUDA_VISIBLE_DEVICES

在 python 命令前加上 CUDA_VISIBLE_DEVICES 可以指定 GPU

CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=0,1,2
CUDA_VISIBLE_DEVICES=2,1,0

你可能感兴趣的:(NVIDIA,cuda)