加上一句:
if torch.cuda:
torch.cuda.set_device(device)
【pytorch】RuntimeError: cublas runtime error 或者 RuntimeError: cublas runtime error
最新推荐文章于 2024-08-01 21:22:04 发布
加上一句:
if torch.cuda:
torch.cuda.set_device(device)