服务器有两块GPU,原本没有指定GPU的,直接用cuda()语句可以运行
今天GPU 0 上面有别人在跑的程序,所以让我指定GPU 1 来跑
一开始试了各种方法代码还是报错,最后简单地加了一条语句就跑通了
import torch
device = 1
torch.cuda.set_device(device)
服务器有两块GPU,原本没有指定GPU的,直接用cuda()语句可以运行
今天GPU 0 上面有别人在跑的程序,所以让我指定GPU 1 来跑
一开始试了各种方法代码还是报错,最后简单地加了一条语句就跑通了
import torch
device = 1
torch.cuda.set_device(device)