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安装教程
解决步骤
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看这一篇就足够了





输入“nvcc -version”或"nvcc -V"时,显示没有nvcc的指令,再次说明没有安装cuda。(忘记截图了)
决定下载CUDA Toolkit 11.3.0。
3.安装CUDA和cuDNN
参考:

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

安装CUDA 11.3.0成功。

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


安装cuDNN v8.2.1成功。
4.卸载pytorch
参考:
在第一步里可以检查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.问题解决

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐



所有评论(0)