广告区

广告区


本站消息

站长简介/公众号

关注本站官方公众号:程序员总部,领取三大福利!
福利一:python和前端辅导
福利二:进程序员交流微信群,专属于程序员的圈子
福利三:领取全套零基础视频教程(python,java,前端,php)

  价值13000svip视频教程,python大神匠心打造,零基础python开发工程师视频教程全套,基础+进阶+项目实战,包含课件和源码

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2021-07(1)

2021-08(13)

关于pytorch与CUDA版本匹配问题

发布于2022-06-20 23:41     阅读(293)     评论(0)     点赞(28)     收藏(1)



项目场景:

在跑CLAM实验时遇到如下问题:
RuntimeError: CUDA error: no kernel image is available for execution on the device


问题描述

使用pytorch训练深度学习模型遇到如上问题。


原因分析:

查看相关资料得知这是由CUDA与pytorch版本不匹配导致的,且查看当前pytorch对应的CUDA版本未10.X,而服务器所用CUDA版本为11.4,因而需要对pytorch版本进行调整。


解决方案:

第一步本人尝试安装最新版本的pytorch
1、登录pytorch官网 https://pytorch.org/get-started/locally/
2、选择如下
第一次所用pytorch安装选项
3、安装后这个问题仍然没有解决

第二步本人决定不再安装稳定版本的用更新的pytorch,于是本人选择如下

在这里插入图片描述
安装完成后问题得到解决


总结

1、该问题是由CUDA版本与pytorch导致的,一般根据CUDA版本调整pytorch即可。
之前版本的pytorch与CUDA的对应关系链接如下
https://pytorch.org/get-started/previous-versions/
2、在pytorch版本不够新的时候可以选择非稳定版本的安装。

原文链接:https://blog.csdn.net/qq_34532413/article/details/124649370







所属网站分类: 技术文章 > 博客

作者:38373

链接:https://www.pythonheidong.com/blog/article/1594364/9974b7d4569928983725/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

28 0
收藏该文
已收藏

评论内容:(最多支持255个字符)