torch 设置不使用GPU torch not compiled with cuda enabled

目录

• 问题
• 查阅资料
• 解决步骤

• 1.检查pytorch版本、是否有CUDA
• 2.安装CUDA前看电脑的显卡驱动程序版本、支持的最高版本
• 3.安装CUDA和cuDNN
• 4.卸载pytorch
• 5.重新安装pytorch
• 6.问题解决

问题

在pycharm上用python3运行代码时报错：

---

查阅资料

报错：Torch not compiled with CUDA enabled看这一篇就足够了

错误Torch not compiled with CUDA enabled解决方法附CUDA安装教程及Pytorch安装教程

CUDA安装教程（超详细）

---

解决步骤

1.检查pytorch版本、是否有CUDA

参考：

错误Torch not compiled with CUDA enabled解决方法附CUDA安装教程及Pytorch安装教程

结论：发现没有安装CUDA，pytorch版本不对：“…+cpu”应为：“…+cu…”。

2.安装CUDA前看电脑的显卡驱动程序版本、支持的最高版本

参考：

错误Torch not compiled with CUDA enabled解决方法附CUDA安装教程及Pytorch安装教程

报错：Torch not compiled with CUDA enabled看这一篇就足够了

CUDA安装教程（超详细）

输入“nvcc -version”或"nvcc -V"时，显示没有nvcc的指令，再次说明没有安装cuda。（忘记截图了）

决定下载CUDA Toolkit 11.3.0。

3.安装CUDA和cuDNN

参考：

CUDA安装教程（超详细）

因为我是第一次下载，所以我这里全选了。

安装CUDA 11.3.0成功。

为了保险起见，我选择和CUDA 11.3相近的一个cuDNN版本：cuDNN v8.2.1。

安装cuDNN v8.2.1成功。

4.卸载pytorch

参考：

关于anaconda 导入第三方工具包的方法

在第一步里可以检查pytorch版本。我这里再通过输入“pip list”检查一下，可以看到有torch包。

输入“pip uninstall torch”，卸载torch。

输入“pip list”，发现没有torch包了。

5.重新安装pytorch

参考：

错误Torch not compiled with CUDA enabled解决方法附CUDA安装教程及Pytorch安装教程

报错：Torch not compiled with CUDA enabled看这一篇就足够了

发现pytorch官网没有CUDA 11.3可以选择，我去搜了历史版本：https://pytorch.org/get-started/previous-versions/

选择了pytorch 1.12.1里对应着CUDA 11.3版本下载。

输入“pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113”下载。

6.问题解决