AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】

问题

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

在这里插入图片描述

  • 4
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值