当出现RuntimeError:CUDA error:no kernel image is available for execution on the device 问题时候的pytorch安装方法

当出现一个明显的特征就是出现:RuntimeError:CUDA error:no kernel image is available for execution on the device 这就说明你的显卡太低了

可以到这个路径下C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite,找到deviceQuenry.exe这个文件拖到cmd命令窗口运行可以看到自身电脑的算力

 从以上可以看到我的电脑算力才3.5,因此这个显卡最多只能支持pytorch1.2

 我在这说一句,大家可以在英伟达控制面板查看自己设备支持安装的最大cuda版本,比如我的显示是cuda11.6那么就是最高不超过安装cuda11.6版本,安装了相应的cuda版本后可以根据以下网址查询相应的pytorch:


在这个网址可以看到你安装的cuda版本下的pytorchhttps://pytorch.org/get-started/previous-versions/

最后根据我的安装经验可以在安装比如cuda11.6版本情况下,选择比cuda11.6低的对应的pytorch,比如安装cuda11.1版本下的pytorch,如下图所示,可以选择11.1版本的pytorch进行安装

 在选择完了要安装的版本后可以到以下网址进行下载安装包,前提要安装wheel库,但一般都会安装。

这个网址你可以通过用whl库进行安装https://download.pytorch.org/whl/torch_stable.html如果下载很慢可以把链接复制到迅雷下载,这样下载速度比较快。下载完成之后可以把文件拖到你想安装的目录下进行安装,一般可以拖到你python运行环境下的目录安装,安装命令如下:

 

最后说一句安装的时候要进入你的文件所在目录的地址,上图的C:\Anaconda3就是我文件所在目录,如果你的文件在其它盘的地方这里以我的D盘做个例子,可以直接在cmd命令窗口输入D: 进入

D盘如下图所示

 之后再输入D盘下的目录地址:CD D:\cuda 如下图所示

 之后就是使用上面提到的pip install torch-1.6.0+cu101-cp36-cp36m-win_amd64.whl 命令进行安装。安装完了torch文件可以把后续的torchvision等一系列文件按照同样的方法进行安装。

安装完成后可以再使用以下代码进行验证

import torch
a=torch.rand(2)
b=a.cuda()
print(b)

 如果没有错误应该得到以下信息:

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

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

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年11月27日
下一篇 2023年11月27日

相关推荐