官方说明:https://www.tensorflow.org/install/
环境:
操作系统 :Windows 10 家庭中文版
处理器 : Intel(R) Core(TM) i7-7700 CPU @3.6GHZ 3.60GHZ
内存 :16GB
显卡:NVIDIA GeForce GTX 1060 6GB
Python:3.6.2
安装GPU版
pip3 install --upgrade tensorflow-gpu
安装 NVIDIA CUDA® Toolkit 8.0.
下载地址
https://developer.nvidia.com/cuda-downloads
安装 cuDNN v6
下载地址:https://developer.nvidia.com/rdp/cudnn-download
需要注册用户
将下载后的zip文件解压到c盘 C:cudnn-8.0-windows10-x64-v6.0
将 C:cudnn-8.0-windows10-x64-v6.0cudain 加入到环境变量Path
验证tensorflow-gpu安装
执行sess = tf.Session()
后,会显示显卡GPU信息。