【深度学习笔记】——win10安装tensorflow-gpu==2.0.0-rc0

目 录

1 安装CUDA10.0（注意不是10.1）和CUDNN for CUDA 10.0

  因为前面安装过tensorflow-gpu 1.13，所以这里不详细介绍了，有兴趣可以看那个帖子。安装2.0最大的不同就是用CUDA10.1无法正确安装，所以先把CUDA都卸载之后再安装10.0的版本

• 卸载tensorflow 1.13

pip uninstall tensorflow-gpu

• CUDA10.0下载地址

• 取消勾选“GeForce Experience”

• 查看“Driver comonents”

  前面的序列号是CUDA种包含的驱动版本，后面的是你计算机中的驱动版本，如果当前版本更高，那么该项也取消勾选

• 可能提示安装Visual Studio

  绝大部分计算机中是有适应的版本的，所以不会提示，如果没有那么可以先安装，另一种可能是计算机中有更高的版本，比如2019，那么直接勾选“忽略该提示”继续安装就可以了

• 安装完毕，检查CUDA成功安装

  安装目录的bin路径下有nvcc.exe

  安装目录的extras/CUPTI/libx64下有cuti64.dll

• CUDNN下载地址

  需要注册一个用户

  选择对应CUDA版本的CUDNN

• 将下载后的文件解压到CUDA的安装路径下

  直接覆盖就可以

• 新增环境变量

  将以下三个变量增加到系统变量的Path中

C:\NVIDIA\CUDAv10.0\bin

C:\NVIDIA\CUDAv10.0\include

C:\NVIDIA\CUDAv10.0\lib\x64

• 测试CUDA

  cmd下运行nvcc -V显示版本为10.0说明成功安装CUDN

2 安装tensorflow

  直接在cmd中运行

pip install tensorflow-gpu==2.0.0-rc0

至此安装完毕，可以在python中运行tf.test.is_gpu_available()，显示True说明成功，也可以跑一个小的程序进行尝试

import tensorflow as tf

tf.__version__