[CUDA_VISIBLE_DEVICES]指定哪张卡运行

使用

在多卡环境下，可以设置环境变量CUDA_VISIBLE_DEVICES来指定要运行的GPU

单张卡

# 设置方式一
CUDA_VISIBLE_DEVICES=0 python xxx.py
# 设置方式二
export CUDA_VISIBLE_DEVICES=0
pytohn xxx.py

多张卡

CUDA_VISIBLE_DEVICES='0,1'