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

目录

  • 问题
  • 查阅资料
  • 解决步骤
    • 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安装教程

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第1张图片

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

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

参考:

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

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

CUDA安装教程(超详细)

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第2张图片

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第3张图片

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第4张图片

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第5张图片

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第6张图片

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

决定下载CUDA Toolkit 11.3.0

3.安装CUDA和cuDNN

参考:

CUDA安装教程(超详细)

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第7张图片

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第8张图片

安装CUDA 11.3.0成功。

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第9张图片

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第10张图片

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第11张图片

安装cuDNN v8.2.1成功。

4.卸载pytorch

参考:

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

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第12张图片

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第13张图片

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第14张图片

5.重新安装pytorch

参考:

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

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第15张图片

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

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第16张图片

选择了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”下载。

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第17张图片

6.问题解决

AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】_第18张图片

你可能感兴趣的:(Python,pycharm,pip,pytorch)