Windows平台
我的环境:tensorflow_gpu 1.9.0 + CUDA 9.0 + cuDNN 7.4.1(建议安装tensorflow_gpu 1.9.0,高版本的可能无法很好使用)
一、首先说一下安装顺序(注意版本一定要匹配,不一定是最新的就能用)
1、安装tensorflow_gpu(注意tensorflow与python版本的匹配)
tensorflow的github仓库:https://github.com/tensorflow/tensorflow
2、根据tensorflow版本查询CUDA版本,安装CUDA
CUDA历史版本下载地址:https://developer.nvidia.com/cuda-toolkit-archive
3、根据CUDA版本查询cuDNN版本,安装cuDNN
cuDNN历史版本下载地址:https://developer.nvidia.com/rdp/cudnn-archive