

[CUDA_VISIBLE_DEVICES]指定哪张卡运行

参考：CUDA Pro Tip: Control GPU Visibility with 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'