出现问题:
检查一下cuda能用吗
其实是因为我这个源代码人家是用的三个gpu跑的而我只有一个
所以解决办法是
在代码的import***所有之后加上
torch.cuda.current_device()
torch.cuda._initialized = True
然后就成功跑通啦

出现问题:
检查一下cuda能用吗
其实是因为我这个源代码人家是用的三个gpu跑的而我只有一个
所以解决办法是
在代码的import***所有之后加上
torch.cuda.current_device()
torch.cuda._initialized = True
然后就成功跑通啦