pytorch指定使用的单个GPU

使用单个GPU

直接在代码中加入下面

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

猜你喜欢

转载自blog.csdn.net/zhuoyuezai/article/details/83007210