当前位置：
  1. 首页
  2. python
  3. 正文
本文介绍: 1.检查pytorch版本、是否有CUDA2.安装CUDA前看电脑的显卡驱动程序版本、支持的最高版本3.安装CUDA和cuDNN4.卸载pytorch5.重新安装pytorch6.问题解决

问题

在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.问题解决

在这里插入图片描述

显示所有内容
上一篇 【pip】raise MaxRetryError(_pool, url, error or ResponseError(cause))
下一篇 pip源更换到国内镜像(cmd)

相关文章

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

插入图片
浏览器会保存昵称、邮箱和网站cookies信息，下次评论时使用。