首页 > Python资料 博客日记
成功解决:AssertionError: Torch not compiled with CUDA enabled
2024-03-12 09:00:03Python资料围观301次
在运行pycharm项目的时候,出现了以上的报错,主要可以归结于以下两个个方面:
1、没有安装GPU版本的pytorch,只是使用清华的镜像地址下载了CPU版本的pytorch
2、安装的CUDA和安装的pytorch的版本不相互对应
我使用 pip list 来查看我在该环境下安装了哪些依赖项,发现自己的torch是CPU版本的,
如何区分的呢?主要通过安装的环境的版本号的后面时候加上了cu,比如说下面这幅图就可以看出这里采用的CUDA版本号为11.3,所以在我的环境里面没有安装GPU版本的torch.
找到问题后,我们就可以进行解决了。
如果自己不是CUDA版本的torch,可以使用pip uninstall torch的命令来卸载已经安装好的torch后继续下面的操作.
第一步:查看自己电脑上安装的python版本,因为python的版本对应者pytorch的版本,pytorch的版本又对应着CUDA的版本。所以先查看自己的python的版本,那么如何查看自己的电脑上的python版本呢?
可以在pycharm的Terminal中,在自己使用的环境下,输入python:下面图中的python 3.6.2就是我的Python的版本号。
如何查看自己项目的环境配置,可以通过pycharm上:
File---Settings---Project--Python Interpreter
如何进入到Terminal,可以通过下面的提示。
第二步:下载离线的torch包。找到python对应的版本,就可以在下面的这个网站中选择现在对应的torch版本,注意我们选择下载torch时要注意python的版本号,并且在选择的时候选择cu开头的链接下载,cu对应的CUDA的版本,比如这里的python版本为3.6,那么只要选择cp36的链接下载
cu102/torch-1.7.1-cp38-cp38-linux_x86_64.whl
- cu102:表示cuda版本为10.2,
- torch-1.7.1:表示torch版本为1.7.1
- cp38:表示适用python版本为3.8
- linux:表示适用于linux系统
- x86_64:表示同时兼容32和64位系统、
第三步:下载对应版本的CUDA。在上面下载torch的时候也可以看到自己对应的CUDA的版本
点击 CUDA下载官网 进入到CUDA的下载官网,往下滑,直到最下面,然后找到下图提示的内容
就可以看到CUDA的很多版本的下载连接,选择我们需要的版本进行下载
选择好下载的版本后,进入下载页面选择我们下载的系统,以及下载的windows系统的版本,和下载方式就可以进行CUDA的下载
下载好CUDA就可以对它进行安装了
点击我们下载好的CUDA的exe程序后,就会出现CUDA的安装页面,这里显示的是安装地址,一般可以不做处理,安装到它默认的地址。
在安装选项中选择自定义安装,然后点击下一步,后面的步骤相同,一直点击(下一步)直到完成CUDA的安装。
在安装完成之后,可以打开系统的环境变量来检查CUDA
是否已经加入到环境变量中了,默认情况下是自动添加到环境变量中的。
检查是否安装成功:打开cmd,输入 nvcc -V, 出现你的CUDA的版本信息就说明安装成功
第四步:将刚才下载好的torch包进行安装,放在项目的目录下:我这里的目录的路径为: D:\zuoye\ReID\deep-person-reid,在pycharm进入到Terminal中,进入到自己的环境中,pip安装对应的torch包,就可以实现GPU版本的torch的安装,也就可以解决上面的问题了。
以上就是这个错误的全部解决方法,总结一下就是:找到对应的python版本,下载python版本对应下的GPU版本的torch文件包,然后下载CUDA,将下载的包放在项目下,然后进行pip安装即可。
标签:
相关文章
最新发布
- 从零到一!超详细Pycharm安装教程(图解+详细步骤)
- python json jsonl 的用法
- 【Python篇】深度探索NumPy(下篇):从科学计算到机器学习的高效实战技巧
- boto3:Python连接S3对象存储并进行文件操作(上传、下载、删除)
- 全网最适合入门的面向对象编程教程:50 Python函数方法与接口-接口和抽象基类
- Python pycryptodome类库使用学习总结
- import torch 报错:WinError 126
- 如何在Java中实现邮件发送功能?
- python之openpyxl模块——实现Excel表格的处理(万字教学,全网最全,超详细!)
- BUG解决:postman可以请求成功,但Python requests请求报403
点击排行
- 版本匹配指南:Numpy版本和Python版本的对应关系
- Python 可视化 web 神器:streamlit、Gradio、dash、nicegui;低代码 Python Web 框架:PyWebIO
- 版本匹配指南:PyTorch版本、torchvision 版本和Python版本的对应关系
- 相关性分析——Pearson相关系数+热力图(附data和Python完整代码)
- Could not build wheels for llama-cpp-python, which is required to install pyproject.toml-based proj
- Python pyinstaller打包exe最完整教程
- Anaconda版本和Python版本对应关系(持续更新...)
- Python与PyTorch的版本对应