当前位置:   article > 正文

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

torch not compiled with cuda enabled

问题

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

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/71839
推荐阅读
相关标签
  

闽ICP备14008679号