安装Tensorflow-gpu版本

1. 下载cuda

链接：https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal



如果电脑上本身就有Visual Studio Integration，要将这个取消勾选，避免冲突了

点开Driver comonents，Display Driver这一行，前面显示的是Cuda本身包含的驱动版本是411.31

如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号，那一定要把这个勾去掉。否则会安装失败(相同的话，就不用去取勾了)



接下来需要几分钟的时间安装，这是NVIDIA程序已完成的界面



打开此路径，查看nvcc.exe

有这个nvcc.exe就说明CUDA安装已成功



打开此文件夹，查看有没有cuti64_100.dll

有这个cuti64_100.dll就说明CUPT1已成功



b. cuDNN 安装

CDNN官网链接：

https://developer.nvidia.com/rdp/cudnn-download

选择cuDNN for CUDA10.0



解压cuDNN



解压会出现三个文件夹：



将这三个文件夹里面的文件对应的复制到cuda文件下：

（注意这里是将文件下的文件复制到cuda对应的文件夹里面，而不是将文件夹直接替代cuda下的文件夹（这步特别重要））







c. PATH配置

查看CUDA环境路径

我的电脑——>属性——>高级系统设置——>环境变量



在系统变量中找到PATH

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

（要是自定义安装路径的话自行修改）

测试CUDA：

cmd

nvcc -V

下面的显示说明我们CUDA的版本是10.0



3. TensorFlow 2.0 安装与测试

a. 确认显卡

安装前要确认显卡是不是NVDIA的

命令行

pip install tensorflow-gpu==2.0.0-beta0



b. 测试

测试Tensorflow是否安装成功：

步骤：

打开cmd——>ipython——>import tensorflow as t f

import tensorflow as tf

tf.test.is_gpu_available()

显示True，说明gpu版本已经安装成功

欢迎加入IT技术交流群：861016679，有问题欢迎留言，一起讨论，进步