pytorch 使用gpu報錯CUDA error: no kernel image is available for execution on the device

報錯原因

cuda版本和 pytorch不匹配

解決辦法

點擊此連接 進入如下圖所示的頁面

查看cuda版本：nvidia-smi

根據自己所需的torch、系統、cuda版本選擇對應的安裝方式即可

查看各版本對應關系

1.這里pytorch和cudatoolkit版本對應關系：
https://pytorch.org/get-started/previous-versions/

2.cudatoolkit版本和系統cuda對應關系：
https://www.zhihu.com/question/344950161/answer/818139888

3.系統cuda和nvidia對應關系：
https://blog.csdn.net/He_9520/article/details/100032803

4.cuda和cuDNN的關系和對應關系：
https://www.jianshu.com/p/622f47f94784
https://www.cnblogs.com/yeran/p/11345990.html

查看顯卡版本：
ubuntu-drivers devices
nvidia-smi

查看服務器的cuda版本
使用如下命令：
nvcc -V

參考鏈接