一. 改变系统环境变量 1.改变系统环境变量仅使目标显卡,编辑 .bashrc文件，添加系统变量： export CUDA_VISIBLE_DEVICES=0 # 这里是要使用的GPU编号 2.在程序开头设置： os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3' # 在运行程序时指定 一般先在程序开头输入该代码。 3.运行程序时使用命令行,来设置该程序可见的gpu: CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py 在终端下
