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

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

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

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

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

你可能感兴趣的:(pytorch,python,pytorch)