Windows环境下的TensorFlow安装过程

Posted 2020-09-21 lllittletree

安装环境

• 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

安装过程

• 安装Python3.5

• 安装python     

     下载地址： https://www.python.org/

1. 点击Downloads下的Windows，本次选用了Python 3.5.2版本

    

      选择其中的第五个，Windows x86-64 executable installer下载安装（安装的时候最好把添加环境变量那个打钩，这样就不需要自己配置环境变量）

     2. 升级pip

        打开cmd输入

python.exe -m pip install --upgrade pip

• 安装TensorFlow

     （本次采用的在线安装，也可以下载到本地，进行本地安装）

     安装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环境变量。

• 测试TensorFlow

打开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(1)
>>> b = tf.constant(2)
>>> print(sess.run(a + b))
3

如遇到‘ is not a supported wheel on this platform’ 问题，请看博客：

tensorflow安装遇到‘ is not a supported wheel on this platform’ 问题

以上内容参考自网络教程，本人以博客的方式记录，如有侵犯请告知。