无法将 GPU 与 Pytorch 一起使用答案

【问题标题】：Can't use GPU with Pytorch无法将 GPU 与 Pytorch 一起使用
【发布时间】：2020-04-17 18:56:16
【问题描述】：

我在尝试使用 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 运行良好。

【问题讨论】：

This 可能会给你一个想法。
你的nvidia驱动是什么版本的？
你的 CUDA 版本是多少？要进行检查，您可以在终端中输入 nvidia-smi。
@xenotecc pytorch 不使用系统 cuda。它安装自己的本地版本
你是对的。抱歉打扰了。

标签： tensorflow machine-learning anaconda artificial-intelligence pytorch

【解决方案1】：

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

【讨论】：

【解决方案2】：

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

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

【讨论】：