1、从nvidia官网下载cuda11.1

https://developer.nvidia.com/cuda-toolkit-archive

2、从nvidia官网下载cudnn8.0

https://developer.nvidia.com/rdp/cudnn-archive

3、下载完打开安装程序,安装路径选择自定义,自定义选项,选择cuda 打勾,然后一路下一步安装完成

 

 

 4、(1)配置环境变量,如下图

(2)检查是不是有图中的两个环境变量

 5、验证安装,如下所示,表示 cuda安装成功

 6、cudnn安装

 



 7、添加cuda11.1下的binincludelib文件夹,复制这个路径到环境变量里

 如下图所示

 

 至些cuda,cudnn配置完成

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐