没有合适的资源?快使用搜索试试~ 我知道了~
首页解决AssertionError Torch not compiled with CUDA enabled.docx
资源详情
资源评论
资源推荐
当运行 pytorch 代码出现报错:
AssertionError: Torch not compiled with CUDA enabled
造成这一错误的原因是 pytorch 没办法调用 cuda(前提是已经安装有 cudatoolkit),根本
原因是计算机的 GPU 驱动与 CUDA 版本不对应
解决的主要方法是调整 CUDA 版本,但是需要注意的是 CUDA 版本又与 pytorch 版本相
对应,所以比较麻烦。
1、需要检查自己的 GPU 所对应的驱动,
打开终端运行:nvidia-smi 可以查看到自己的版本
2、结合自己的 GPU 版本(如 NVIDIA-SMI 384.130)选择合适的 CUDA 和 CUDNN,
如果 GPU 驱动是 384 时,选择 CUDA8.0 和 CUDNN6.0 比较合适,当然如果驱动越高
CUDA 的版本也会越高,但是不要盲目按照高版本,会导致无法查找到相应的 CUDA
3、安装 CUDA8.0 和 CUDNN6.0,最简单的方法就是在 anaconda navigator 中将已安
装的 cudatoolkit 和 cuDNN 降到 8.0 和 6.0,当然还有其他的安装方式可以参考
4、调整 pytorch 版本,由于 cudatoolkit 对应的版本低于 0.4,可以选择将 pytorch 版
本降到 0.2,调整方法也可采用 anaconda navigator。
5、调整完 pytorch 版本后,还需要调整其他相应的包,如 torch,
6、检查是否解决问题;
进入 python 环境
输入:import torch
Torch.cuda.is_available()
如果输出为:ture,则表明已经解决上述问题。
临海观风
- 粉丝: 1
- 资源: 28
上传资源 快速赚钱
- 我的内容管理 收起
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
会员权益专享
最新资源
- RTL8188FU-Linux-v5.7.4.2-36687.20200602.tar(20765).gz
- c++校园超市商品信息管理系统课程设计说明书(含源代码) (2).pdf
- 建筑供配电系统相关课件.pptx
- 企业管理规章制度及管理模式.doc
- vb打开摄像头.doc
- 云计算-可信计算中认证协议改进方案.pdf
- [详细完整版]单片机编程4.ppt
- c语言常用算法.pdf
- c++经典程序代码大全.pdf
- 单片机数字时钟资料.doc
- 11项目管理前沿1.0.pptx
- 基于ssm的“魅力”繁峙宣传网站的设计与实现论文.doc
- 智慧交通综合解决方案.pptx
- 建筑防潮设计-PowerPointPresentati.pptx
- SPC统计过程控制程序.pptx
- SPC统计方法基础知识.pptx
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论1