CUDA、cuDNN以及pytorch_GPU的安装与配置

win10+cuda+cuDNN+pytorch1.4.0

---

首先先说一下，由于cuda下载需要与本地的显卡驱动程序相匹配，所以先检查一下自己的驱动程序属什么版本，版本太低的话可以先更新一下（pytorch1.4目前只支持cuda9.0以上版本，所以显卡驱动程序版本太低也不行）

 

                                           

打开显卡驱动程序，系统信息中检查NvCUDA.DLL的版本。下面是不同版本所对应的cuda版本

 

                           

显卡驱动程序更新方法如下：鼠标右键单击计算机-选择管理-在系统工具中-点击设备管理器-展开显示适配器-选中显卡驱动程序-右键单击它-选择更新驱动程序-弹出窗口-选择自动搜索更新选项-待安装完成即可。好接下来可以进行CUDA和pytorch_GPU的配置了。

1、先下载安装CUDA

下载地址：https://developer.nvidia.com/cuda-downloads，选择相应版本的CUDA进行下载就行（我选择的是本地安装，也就是先把安装程序下载下来，