关于pytorch与CUDA版本匹配问题

项目场景:

在跑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与CUDA版本匹配问题
3、安装后这个问题仍然没有解决

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

关于pytorch与CUDA版本匹配问题
安装完成后问题得到解决

; 总结

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

Original: https://blog.csdn.net/qq_34532413/article/details/124649370
Author: DragonJ__
Title: 关于pytorch与CUDA版本匹配问题

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/705837/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球