一、安装 cuda 9.0
下载 cuda 9.0 (windows 10 x64版本): https://developer.nvidia.com/cuda-toolkit-archive
按以下顺序进行下载安装:
检查安装结果:
环境变量已自动设置好了,不用手动设置。
二、安装 cudnn 7.0
下载 cudnn 7.0:https://developer.nvidia.com/rdp/cudnn-archive
(需要注册一个帐号)
下载后解压,拷贝到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA 目录下。
四、安装 Python 3.6 + TensorFlow GPU
目前在Python 3.7 下安装 TensorFlow GPU 失败。转在 Python 3.6 下安装 TensorFlow GPU成功。
pip install tensorflow-gpu
五、检查 TensorFlow GPU 安装情况
进入 Python 3.6:
没有出错信息即表示安装成功了!