关于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安装选项
3、安装后这个问题仍然没有解决

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


安装完成后问题得到解决


总结

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

来源:DragonJ__

物联沃分享整理
物联沃-IOTWORD物联网 » 关于pytorch与CUDA版本匹配问题

发表评论