zhuzhishi0 2019-06-21
(下载cuDNN需要先注册NVIDIA开发账户并登录才能看到下载界面)
CUDA 9 要TF 1.5版本才支持
注意windows tf 仅支持 python 3.5
为了避免之后使用 pip 安装 tf 报错,请先在cmd运行以下代码 pip install --ignore-installed --upgrade pip setuptools
之后安装tfpip install tensorflow-gpu==1.1.0
接来下检验安装是否成功
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))另外,如果用conda的话一定要注意python版本,目前最高支持3.5
conda create --name tensorflow python=3.5 activate tensorflow pip install tensorflow-gpu
原文作者: filwaline