目录
一、安装Visual Studio
根据版本,这里需要安装Visual Studio Community2015 版本(具体版本号对应看后面的图)。
具体见我的上一篇文章:https://blog.csdn.net/mango_ZZY/article/details/111631900
二、安装CUDA
1.查看自己NVIDIA版本
①打开cmd,然后分别输入如下命令:
# 进入该文件夹
cd C:\Program Files\NVIDIA Corporation\NVSMI
#查看版本
nvidia-smi
我的版本为391.25
2.找到相对应的CUDA版本
根据如下步骤查看CUDA版本
①右键“我的电脑”——>属性——>控制面板主页
②点击“硬件与声音”——>NVIDIA控制面板
③点击:帮助——>系统信息——>组件
可以看到版本信息,我的NVIDIA对应的CUDA最高版本是9.1.84。
3.下载CUDA
这里给出CUDA历史版本的下载网址 https://developer.nvidia.com/cuda-toolkit-archive ,需要找到对应比的版本号进行下载。
官网中9.1版本的只有9.1.85,因此,我要下载9.0版本(9.0.176)。
安装类型有两种:在线安装和本地安装,这里推荐选择local,如果网速快可以选择在线安装。
三、安装CUDNN
官网给出了对应关系:https://tensorflow.google.cn/install/source_windows#gpu