conda安装pytorch总是仅支持cpu
时间: 2023-08-07 08:11:42 浏览: 254
快速安装Pytorch以及CUDA匹配问题
5星 · 资源好评率100%
如果你在使用 `conda` 安装 PyTorch 时发现只支持 CPU,可能是因为你的计算机上没有安装 NVIDIA 的 GPU 驱动程序或 CUDA 工具包。如果你想使用 GPU 版本的 PyTorch,你需要确保你的计算机上有 NVIDIA 显卡并已经正确安装了对应的驱动程序和 CUDA 工具包。
你可以通过以下命令检查你的计算机上是否安装了 NVIDIA 显卡驱动程序:
```
nvidia-smi
```
如果你看到了 NVIDIA 显卡的信息,则说明驱动程序已经成功安装。如果你没有看到任何信息,则需要先安装 NVIDIA 显卡驱动程序。
如果你已经安装了 NVIDIA 显卡驱动程序和 CUDA 工具包,但仍然无法安装 GPU 版本的 PyTorch,可以尝试使用 `pip` 安装,即:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
上面的命令将安装 CUDA 11.1 版本的 PyTorch。如果你安装了其他版本的 CUDA,可以根据需要替换 URL 中的 `cu111`。
阅读全文