Windows环境下安装TensorFlow并在Jupyter notebook上使用

时间:2022-02-11 22:47:23

我的系统是 Windows 10 64-bit,安装了Python 3.5。

TensorFlow 官网给出了五种安装方法,之前r1.0出来后我就给win10装上了,用的是最简单的pip方法。看到别人用 Jupyter notebook 运行代码,感觉很高大上的样子,于是也想装一个用用。

具体过程在第二部分,用Anaconda安装TensorFlow。


用pip方法安装TensorFlow

  1. 先在电脑上装一个Python,注意要装TF支持的Python版本。
    Python 3.5.x from python.org

    TensorFlow only supports version 3.5.x of Python on Windows.

  2. 打开终端。装个pip3包管理器,装TF要用,装很多库也可以用它。
    (装CPU还是GPU版本 参照TF官网windows安装的说明查下显卡即可)

    C:\>pip3 install --upgrade tensorflow
    C:\>pip3 install --upgrade tensorflow-gpu

安装成功以后,直接在Python自带的IDLE上就可以使用了。
注意:用户创建的程序脚本XX.py,不能用tensorflow.py作为文件名,会产生冲突出错。


用Anaconda安装TensorFlow

  1. Anaconda官网下载,先安装好Anaconda,我之前装过的,所以从下面开始。

  2. 打开Anaconda的终端,这个
    Windows环境下安装TensorFlow并在Jupyter notebook上使用

  3. 建立一个 conda 计算环境,命名为tensorflow

    D:\>conda create -n tensorflow python=3.5

    由于Anaconda内置的Python版本是3.6,而此时的tensorflow只支持Python 3.5,这里最好设定一下python版本,让这个conda环境专门为tensorflow服务。

    否则下一步可能会报错说该平台该环境不支持TF的安装;即使不出错地装上了python3.5的TF,在第6步装ipython时又会跳坑…

    ——如果建错了python版本的conda环境,删除环境方法见文末附1。

  4. 激活这个环境,使用 conda 安装 tensorflow (同样可选CPU或GPU版本)

    D:\>activate tensorflow
    (tensorflow)D:\> # Your prompt should change

    (tensorflow)D:\>pip3 install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.whl
  5. 先测试下是否成功,在终端进入 python,输入 import tensorflow as tf 看看TensorFlow 安装成功没

    (tensorflow)D:\>python
    ···
    >>>import tensorflow as tf
    >>>hello = tf.constant("hello, tensorflow!")
    >>>sess = tf.Session()
    >>>print(sess.run(hello))
    hello, tensorflow!

    安装成功后,每次使用 TensorFlow 的时候都需要激活 conda 环境。
    查看环境,激活环境,关闭环境依次是:

    D:\>conda info --envs

    D:\>activate tensorflow
    (tensorflow)D:\>

    (tensorflow)D:\>deactivate tensorflow
    D:\>
  6. 以上算是Anaconda安装好了TensorFlow,但要想在Jupyter notebook上使用,还没完
    接着安装ipython,安装jupyter。

    (tensorflow)D:\>conda install ipython
    ···
    (tensorflow)D:\>conda install jupyter
    ···

    没错就是这里… 创建tensorflow这个conda环境时不设定python版本的话,因为Anaconda内置的python版本是3.6,安装ipython时就会有这一项,并且所有包都基于py36:
    Windows环境下安装TensorFlow并在Jupyter notebook上使用
    于是这个专门为TF服务的conda环境内的python就被更新成了3.6,于是我们前两步刚装好的TF就不能用了,于是只能把这个环境删掉重来…

  7. 输入这个:

    (tensorflow)D:\>ipython kernelspec install-self --user

    看到类似这个结果

    Installed kernelspec python3 in C:\Users\XXX\Jupyter\kernels\python3

    然后再去 jupyter notebook 新建一个文件,试试 import tensorflow as tf,这时应该就可以用啦。

  8. 终于over!!
    Windows环境下安装TensorFlow并在Jupyter notebook上使用

附1. 删除已建的conda环境,该环境必须处于未激活状态,才能删除

(tensorflow)D:\>deactivate tensorflow
D:\>conda remove -n tensorflow --all

D:\>conda info --envs
···
再看一下还有的conda环境,tensorflow那个环境没有啦。

若还是不行的话可以再参考这篇博客 如何在Jupyter Notebook中使用Tensorflow,它是博主翻译自*上的一个问题解答,但原问题背景并不是window环境,所以能不能用我也不太清楚。