关于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版本不够新的时候可以选择非稳定版本的安装。