windows安装cuda以及cudnn
windows安装cuda以及cudnn
·
-
查看自己电脑显卡驱动对应的cuda版本
进入NVIDIA控制面板==>帮助==>系统信息==>组件,找到下图,即显卡对应cuda版本
-
下载cuda程序,下为官网地址,找到对应的CUDA应用程序
https://developer.nvidia.com/cuda-toolkit-archive
-
安装CUDA
① 默认 ok
② 默认 同意并继续
③ 选择自定义
④选择CUDA,其他的都取消,CUDA中把Visual Studio Integration取消I(如果电脑中安装有可以取消勾选),点击 下一步等待安装完成;
⑤安装位置,一般选择默认C盘
如果需要修改,需要创建两个文件夹如下图:
NVIDIA Corporation NVIDIA GPU Computing Toolkit
⑥点击下一步,等待安装完成
-
安装cudnn,cudnn官网,需要邮箱登陆注册:
https://developer.nvidia.com/rdp/cudnn-download
找到对应的cudnn,cudnn下载下来是一个压缩包,类似于CUDA的补丁,下载后解压:
上面文件全选,复制到以下文件夹下(CUDA安装目录)E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3
-
加入系统变量
把以下文件夹路径加入到系统变量Path中E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3\include E:\cudainstall\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp E:\cudainstall\NVIDIA Corporation\CUDA Samples\v11.3\common\lib\x64 E:\cudainstall\NVIDIA Corporation\CUDA Samples\v11.3\bin\win64
-
测试是否安装成功,打开cmd,输入nvcc -V
安装成功
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)