基于Pytorch的深度学习Win10环境配置

CUDA & CUDNN 安装与验证

CUDA安装

查看系统的GPU支持的CUDA版本

nvidia-smi

上图中红色方框框出的为当前系统支持的CUDA最高版本。接下来根据得到的版本信息开始下载CUDA相关安装包。
英伟达官方CUDA下载连接：https://developer.nvidia.cn/cuda-downloads
页面上方确认当前下载的版本：

按照下图红框依次

一般官方页面显示的为最新CUDA环境，历史版本下载链接：https://developer.nvidia.com/cuda-toolkit-archive，页面如下所示，根据当前系统自行选择。

以安装CUDA11.8.0为例：

1. 选择CUDA解压路径（与实际安装位置无关，安装后的解压文件会自动删除），下图是安装包解压路径配置和安装进程图。
   
   
2. 之后开始安装过程，初始界面点击“同意并继续”，接着自定义安装。