两个项目用的一个conda环境,但是torch.cuda.is_available()在一个项目中一直是false,而在另外一个项目中是true
我服了,不知道咋回事
只需要在代码前面加上:
torch.cuda.current_device() torch.cuda._initialized = True
torch.cuda.current_device()
torch.cuda._initialized = True
关于win10环境下在使用torch调用cuda()失败的原因_torch_cuda.dll-CSDN博客
京公网安备 11010502049817号