Windows GPU版本的深度学习环境安装

本文记录了cuda、cuDNN的安装配置。

参考文章:

cuda-installation-guide-microsoft-windows 12.1 documentation

Installation Guide :: NVIDIA cuDNN Documentation

一、cuda安装

注意事项:

1、cuda安装最重要的是查看自己应该安装的版本。

表格数据来源:CUDA 12.1 Release Notes

Windows GPU版本的深度学习环境安装_第1张图片

 2、安装时不要勾选Visual Studio Integration

Windows GPU版本的深度学习环境安装_第2张图片

 二、cuDNN安装

1、选择与cuda对应版本。

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

2、文件合并

下载是个压缩包,解压后里面有bin、include、lib 三个文件夹。

分别把这三个文件夹中的文件合并到cuda的安装目录中。

我本地的是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

Windows GPU版本的深度学习环境安装_第3张图片

你可能感兴趣的:(外行人尝试AI,windows,深度学习,cuda,cuDNN)