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

参考链接


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM