在 python 命令前加上 CUDA_VISIBLE_DEVICES
可以指定 GPU
CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=0,1,2
CUDA_VISIBLE_DEVICES=2,1,0
在 python 命令前加上 CUDA_VISIBLE_DEVICES
可以指定 GPU
CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=0,1,2
CUDA_VISIBLE_DEVICES=2,1,0