无法将 GPU 与 Pytorch 一起使用

Question

我在尝试使用 Pytorch 时不断收到此错误。

运行时错误：尝试反序列化 CUDA 设备上的对象，但 torch.cuda.is_available(( 为 False。如果您在仅使用 CPU 的机器上运行，请使用 torch.load 和 map_location=torch.device('cpu'( 将存储映射到 CPU。

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

使用张量流，我的GPU运行良好。

Answer 1

您可以通过安装 CUDA 10.2(最新版本(来修复此错误，然后使用以下命令重新安装 Pytorch：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

Answer 2

使用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