在代码import torch后加上
torch.cuda.current_device()
即可解决问题
cuda()报错
最新推荐文章于 2023-11-21 15:35:23 发布
在代码import torch后加上
torch.cuda.current_device()
即可解决问题