os.environ[‘CUDA_VISIBLE_DEVICES‘]= ‘0‘设置环境变量

os.environ[‘环境变量名称’]=‘环境变量值’ #其中key和value均为string类型

import os
os.environ["CUDA_VISIBLE_DEVICES"]=‘6‘’，‘7’

这个语句的作用是程序可见的显卡ID。

注意
如果"CUDA_VISIBLE_DEVICES" 书写错误，也不报错，自动选用第一个显卡

版权声明：本文为weixin_43402775原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。