需要的软件
- Win10系统
- NVIDIA驱动
- CUDA10.0
- cuDNN7.6.5
- tensorflow-gpu2.0.0的安装
我提前安装好了Anaconda3(关于Anaconda3和Python的版本–点击),我自己用的是Python3.6。Pycharm2019.3–点击, win10系统就不废话了
NVIDIA驱动
首先你先看一下自己的电脑中有没有英伟达驱动,我的电脑中是原来就配置好的,桌面右键如果菜单栏中有NVIDIA控制桌面如下图,点开查看配置信息。
进入界面,点击左下角–系统信息,会弹出如下的界面,重点看版本号,我的是432.00,对于tensorflow-gpu2.0.0,驱动程序版本有要求,一般大于418就可。如果你的版本过低,需要重新安装驱动。
如果你的电脑没有安装英伟达驱动,安装驱动前一定要看自己电脑的显卡配置,我的电脑–管理–设备管理–显示适配器,下图是我的电脑配置。
NVIDIA驱动,一定要根据自己的电脑的显卡进行选择,下图是我的电脑配置。关于下载类型选择SD还是GRD,跑深度学习选择SD就可以了,游戏的话选择GRD,然后点击–搜索。
会出来如下图所示,点击下载即可。安装过程,网上很多,在这不详细说了。
CUDA10.0
因为要安装TensorFlow-gpu-2.0.0,需要的版本是CUDA10.0(不要用其他版本,可能会报错),CUDA10.0如下图所示
根据自己的需求选择,可以选择在线安装exe,如果你的网速够快。
我直接点击的local,网速太差了,下载完如下图所指示。在双击安装的时候,一定要断网安装,我一开始没有断网安装,显示找不到路径,当然如果你们联网没有出现这种情况最好了。
安装路径默认就可以了,下面是别人的截图,跟我差不多,点击下一步,会出来三个文件夹,应该选其中一个就可以,但我没有管它,直接下一步了,结果导致最终三个文件夹中都有cuda的安装。如果你不放心,你也可以不选择,直接点击下一步。
这一步中我只选择了cuda,其他的都没有选。
选择下面的I understand,等待安装就可以了。安装结束后会出现一个界面,那个界面的两个小框选择不选择都可以,软件的一个说明而已。
cuDNN安装
cuDNN,需要注册,很简单了。注册完下载cuda对应的版本就可以了,如下图,也可以来我的网盘提取–链接:https://pan.baidu.com/s/12-UlAm7t6kG9GE3ss2tJ0Q 提取码:bjq8
下载完是一个压缩包的形式
首先你得解压,把解压文件中的–cuda,复制到如下图所在的位置
添加环境变量,添加的过程不再详说了,大家肯定都会啦
安装完成了,撒花~~~~~~~~~
测试安装的是否正确
用cmd命令测试:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite(如果自定义了安装路径需要修改该路径)中的 bandwidthTest.exe 和 deviceQuery.exe 文件,如果最后都显示 Result = PASS 则说明安装成功。如下图
说明安装成功了
或者在cmd中查看cuda版本
nvcc -V
安装TensorFlow-gpu2.0.0
- 激活环境,我用的Anaconda3中配置好的环境,一定要激活,不然你都不知道自己下到哪里去了
activate base
2.更新pip
python -m pip install --upgrade pip
3.下载安装,用的是镜像版
pip install tensorflow-gpu==2.0.0 -i http://mirrors.aliyun.com/pypi/simple --trusted-host mirrors.aliyun.com
4.查看是否安装完成,如下图
conda list
然后重启电脑就可以啦~~~~~~~~
码字不易,希望点个赞再走吧~~~~~~,后面持续跟新linux系统下的深度学习的配置。。。。。。