print(torch.cuda.is_available()) False

pytorch无法使用cuda。print(torch.cuda.is_available()) 结果为False。  
原因
pytorch1.5与cuda10.0不匹配  
解决方法
在pip清华源​​​​​​​查找自己cuda对应
的torch和torchvision对应的版本进行安装。  
安装指令
 

pip install torch===1.4.0 torchvision===0.5.0


安装完以后。print(torch.cuda.is_available()) 结果为True。

你可能感兴趣的:(pytorch)