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

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

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

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