无法在Pytorch中使用GPU

Posted 2021-04-11

[尝试使用Pytorch时，我一直收到此错误。

RuntimeError：尝试在CUDA设备上反序列化对象，但是torch.cuda.is_available（）为False。如果您在仅CPU的计算机上运行，​​请使用带有map_location = torch.device（'cpu'）的torch.load将存储映射到CPU。

我使用conda install pytorch torchvision cudatoolkit=10.1 -c pytorch安装了Pytorch。

使用tensorflow，我的GPU运行良好。

答案

您可以通过安装CUDA 10.2（最新版本）来解决此错误，并另外使用以下命令重新安装Pytorch：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

另一答案

使用pytorch.org中的官方安装命令>

对于Windows + GPU（对于其他，请检查pytorch.org）

pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html