安装CUDA，CUDNN和tensorflow-gpu涉及的一些网站

经过测试的构建配置，查看CUDA、CUDNN、Tensorflow、python版本匹配

点击查看：https://www.tensorflow.org/install/source_windows?hl=zh-cn

下载CUDA

CUDA地址：https://developer.nvidia.com/cuda-toolkit-archive

下载CUDNN

CUDNN地址：https://developer.nvidia.com/rdp/cudnn-archive

安装教程

https://zhuanlan.zhihu.com/p/416712347

查看tensorflow是否为GPU版本,并检查gpu是否可用

import tensorflow as tf
tf.config.list_physical_devices('GPU')

import tensorflow as tf
tf.test.is_gpu_available()

Could not locate *.dll

在训练模型时总是会报错，例如Could not locate zlibwapi.dll或者Could not load library cudnn_cnn_infer64_8.dll.，需要下载*.dll放在lib文件或者C:\Windows32，其中zlibwapi包下载官网地址。

【Tensorflow】gpus = tf.config.experimental.list_physical_devices(device_type=‘GPU’)

tf.config.experimental.set_memory_growth(gpus[0], True)
参考解决博客

import tensorflow as tf

gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)

经过一天的折腾终于让自己的GPU动起来了。

如果CUDA和CUDNN都安装好，可检查到GPU设备，但是训练不起来，很可能时内存不够，缩小batch_size。或许就成功了。