|
安装环境
- Windows8.1
- python3.5.x(TensorFlow only supports version 3.5.x of Python on Windows)
- pip 9.0.1
- tensorflow-gpu-0.12.0
- cuda
cudnn
安装过程
下载地址: https://www.python.org/
- 点击Downloads下的Windows,本次选用了Python 3.5.2版本
选择其中的第五个,Windows x86-64 executable installer下载安装(安装的时候最好把添加环境变量那个打钩,这样就不需要自己配置环境变量,后续的需要的地方都打钩,一路next)
2. 升级pip
打开cmd输入
python.exe -m pip install --upgrade pip
(本次采用的在线安装,也可以下载到本地,进行本地安装)
安装gpu版本
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0-cp35-cp35m-win_amd64.whl
安装CPU版本
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.0-cp35-cp35m-win_amd64.whl
查看你的gpu是否支持cuda
gpu版本要安装gpu支持包(cuda8.0地址和cudnn5.1地址)
安装cuda8.0
安装对应的cudnn(这个貌似需要登录Nvidia账号才能够下载,我是先注册的)
找到你需要的版本,下载cudnn压缩文件后进行解压,添加解压后的bin目录到PATH环境变量。
打开cmd,输入python,然后输入 import tensorflow,如果结果和下图一样,则说明安装成功。
$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42 |
|
|
|
|
|
|