【报错解决】CUDA error: no kernel image is available for execution on the device

1.查看当前cuda版本

nvcc -V

2.查看当前pytorch 版本

import torch

print(torch.__version__)
print(torch.version.cuda)

4.查看cuda与pytorch匹配的版本

3.降低pytorch 到cuda 匹配的版本

下载地址。点击要的版本,下载到本地。下载后直接拖进工程里了

安装前先卸载 torch。然后在顶级目录安装

pip uninstall torch
pip install torch-1.5.0+cpu-cp38-cp38-linux_x86_64.whl

4.验证是否安装成功

import torch

print(torch.__version__)
print(torch.version.cuda)

应该打印出来了你刚刚装的版本了

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
上一篇 2022年5月18日 下午1:31
下一篇 2022年5月18日 下午1:35

相关推荐

本站注重文章个人版权,不会主动收集付费或者带有商业版权的文章,如果出现侵权情况只可能是作者后期更改了版权声明,如果出现这种情况请主动联系我们,我们看到会在第一时间删除!本站专注于人工智能高质量优质文章收集,方便各位学者快速找到学习资源,本站收集的文章都会附上文章出处,如果不愿意分享到本平台,我们会第一时间删除!