Tensorflow-gpu 1.15安装

CUDA和cuDNN安装

装Tensorflow之前要确保本电脑有CUDA和cuDNN，没有的话用如下方法安装，用官网网址下载较慢，推荐使用Free Download Manager下载器下载

Tensorflow 与cuda的对应关系

cuda与显卡驱动对应关系

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

CUDA下载网址

https://developer.nvidia.com/cuda-toolkit-archive

双击安装先会安装一个安装程序。安装完成后会自动添加环境变量。

cuDNN下载网址

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

cuDNN操作

将cudnn目录下的文件移入到对应的cuda文件夹中（例：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 cudnn.lib->cuda.lib）

tensorflow安装

pip install tensorflow-gpu==1.15

测试

>> import tensorflow as tf
>> tf.test.is_gpu_available()  # 返回True则安装成功