多显卡的分卡使用

Linux系统下

CUDA_VISIBLE_DEVICES=0 python trainer.py 

Windows系统下

（1）代码指定

import os
os.environ["CUDA_VISIBLE_DEVICES"]='0'

（2）环境指定

# 设置临时变量（所有在命令行下对环境变量的修改只对当前窗口有效，不是永久性的修改）
set CUDA_VISIBLE_DEVICES=0

或者是打开系统变量直接设置永久性变量：