简介:这个代码什么意思 os.environ['CUDA_VISIBLE_DEVICES'] = '0, 1' 这段代码是用来设置环境变量的。具体来说,它将CUDA_VISIBLE_DEVICES这个环境变量设置为'0, 1',表示只使用GPU设备0和1来运行程序。 在深度学习中,通常会使用CUDA来加速模型的训练和推理过程,而CUDA_VISIBLE_DEVICES就是用来指定程序可以使用...
os.environ['CUDA_VISIBLE_DEVICES']='0,1,2' # 这句话意思是对于torch来说,只有gpu0,gpu1,gpu2三个gpu是可见的,gpu3不可见os.environ['CUDA_VISIBLE_DEVICES']='1,2' 对于torch来说可见的只有1号和2号卡,并且gpu1为主卡注意“os.environ[……]”这句话要放在import os后、import torch前,即:impor...