numba CUDA报错的问题解决

先打开cmd控制面板,输入nvcc --version,确认电脑上没有cuda

打开nvidia控制面板,看到显卡型号是3050

在帮助–系统信息下打开弹窗,在“组件”下找到CUDA版本,可以看到要下载的是11.4.1版本的

打开nvidia的官方cuda下载地址,找到11.4.1下载

选择深绿色的部分,开始下载

下载后按照默认设置安装即可,其中解压地址可以随便选,安装完之后会把解压的中间文件删除。

重启之后,在cmd中输入命令nvcc --version,发现安装成功

接下来吧cuda添加进环境变量里,用windows自带的搜索可以打开

在“系统变量”下选择“path”,然后点编辑

添加以下两条:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64

这样,就可以正常使用CUDA了

作者:waiting不是违停原文地址:https://blog.csdn.net/weixin_44737922/article/details/123129392

%s 个评论

要回复文章请先登录注册