Python 官方文档:入门教程 => 点击学习
出现问题(已解决)RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker
为什么把警告po出来,是因为警告可以让我们了解一些有用信息。
首先警告里的内容不可忽略,翻译过来就是NVIDIA RTX GeForce 3060Ti(我使用的服务器)支持的CUDA的算力为8.6,与当前的PyTorch的版本不匹配。说白了就是CUDA和pytorch版本不一致。
当前的pytorch版本支持的CUDA的算力为3.7、5.0、6.0、7.0
我安装的cuda是11.3,对应的pytorch版本为1.10.1,但是为了跑这个代码,我又安装了pytorch1.10,这就导致了CUDA和pytorch版本不一致
因此我重新安装cuda11.1,然后在pytorch官网找到对应的pytorch1.10的安装语句
pip install torch==1.10.0+cpu torchvision==0.11.0+cpu torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
现在风扇已经在转了!
pytorch官网 链接: https://pytorch.org/get-started/previous-versions/
深度学习环境配置避坑: https://blog.csdn.net/weixin_46065198/article/details/125642546
来源地址:https://blog.csdn.net/m0_48664152/article/details/126459836
--结束END--
本文标题: 出现错误(已解决)RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA ker
本文链接: https://lsjlt.com/news/404961.html(转载时请注明来源链接)
有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
2024-03-01
2024-03-01
2024-03-01
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
回答
回答
回答
回答
回答
回答
回答
回答
回答
回答
0