- 安装 VisualStudio2017 选择 C++组件安装。
- 选择对应的 CUDA 工具包版本下载安装（一定要符合官方文档指定的版本）。打开 CMD 输入
nvcc -V验证安装。 - 下载对应的 cuDNN（需要注册登陆），解压后分别将
bin、include、lib中的文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0（默认安装目录）下的bin、include和lib中。 - 下载并安装 python3.6.x。
- CMD 输入
pip3 install --upgrade tensorflow-gpu安装 GPU 版的 TensorFlow。 - 验证安装，进入 Python 交互式环境，输入：
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
结果如下：
Hello, TensorFlow!
网友评论