关于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、选择如下
3、安装后这个问题仍然没有解决
第二步本人决定不再安装稳定版本的用更新的pytorch,于是本人选择如下
安装完成后问题得到解决
总结
1、该问题是由CUDA版本与pytorch导致的,一般根据CUDA版本调整pytorch即可。
之前版本的pytorch与CUDA的对应关系链接如下
https://pytorch.org/get-started/previous-versions/
2、在pytorch版本不够新的时候可以选择非稳定版本的安装。
来源:DragonJ__