目 录
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
-
取消勾选“GeForce Experience”
-
查看“Driver comonents”
前面的序列号是CUDA种包含的驱动版本,后面的是你计算机中的驱动版本,如果当前版本更高,那么该项也取消勾选
- 可能提示安装Visual Studio
绝大部分计算机中是有适应的版本的,所以不会提示,如果没有那么可以先安装,另一种可能是计算机中有更高的版本,比如2019,那么直接勾选“忽略该提示”继续安装就可以了
- 安装完毕,检查CUDA成功安装
安装目录的bin路径下有nvcc.exe
安装目录的extras/CUPTI/libx64下有cuti64.dll
需要注册一个用户
选择对应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__