问题
查阅资料
报错: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。(忘记截图了)
3.安装CUDA和cuDNN
参考:
安装CUDA 11.3.0成功。
为了保险起见,我选择和CUDA 11.3相近的一个cuDNN版本:cuDNN v8.2.1。
4.卸载pytorch
参考:
在第一步里可以检查pytorch版本。我这里再通过输入“pip list
”检查一下,可以看到有torch包。
输入“pip uninstall torch
”,卸载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.问题解决
原文地址:https://blog.csdn.net/PLCET/article/details/127875993
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若转载,请注明出处:http://www.7code.cn/show_32628.html
如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱:suwngjj01@126.com进行投诉反馈,一经查实,立即删除!