Win10下搭建深度学习环境

目录

一、安装Visual Studio

二、安装CUDA

1.查看自己NVIDIA版本

2.找到相对应的CUDA版本

3.下载CUDA

三、安装CUDNN

四、安装TensorFlow-GPU

---

一、安装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

 

四、安装TensorFlow-GPU

https://blog.csdn.net/weixin_41035425/article/details/108397853?ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522160878993516780265398471%252522%25252C%252522scm%252522%25253A%25252220140713.130102334..%252522%25257D&request_id=160878993516780265398471&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~baidu_landing_v2~default-3-108397853.first_rank_v2_pc_rank_v29&utm_term=visual%20studio%E6%B7%B1%E5%BA%A6%E5%AD%A6%E4%B9%A0