加上一句:
if torch.cuda:
torch.cuda.set_device(device)
【pytorch】RuntimeError: cublas runtime error 或者 RuntimeError: cublas runtime error
最新推荐文章于 2025-06-04 13:11:02 发布
加上一句:
if torch.cuda:
torch.cuda.set_device(device)