Linux系统下 CUDA_VISIBLE_DEVICES=0 python trainer.py Windows系统下 (1)代码指定 import os os.environ["CUDA_VISIBLE_DEVICES"]='0' (2)环境指定 # 设置临时变量(所有在命令行下对环境变量的修改只对当前窗口有效,不是永久性的修改) set CUDA_VISIBLE_DEVICES=0 或者是打开系统变量直接设置永久性变量: