『方法』Anaconda环境下安装TensorFlow

Author Avatar
IDforHYIT 3月 10, 2019

本页面访问量

  • 在其它设备中阅读本文章

『方法』Anaconda环境下安装TensorFlow

还在使用 pip 安装 TensorFlow?试试 conda 吧,简单方便安装 TensorFlow,且能带来更快的 CPU 性能。

TIPS

下列命令行坏境,如未加入系统环境变量,均需切换到对应目录后在文件管理器地址栏敲上cmd后回车。

目录参考

D:\envs\Python\Anaconda3\Scripts

1. 卸载原版本

如果之前安装过的话,请先卸载

pip uninstall tensorflow

2. conda

conda install tensorflow

此外,如果你想要 GPU 版本,那么将上面一行中的 tensorflow 换成 tensorflow-gpu。

补充:

当然,Anaconda发行网站推荐你创建一个虚拟环境来放置TensorFlow,这样的话可以避免潜在的冲突,命令修改如下:

CPU版本:

conda create -n tensorflow_env tensorflow
conda activate tensorflow_env

GPU版本:

conda create -n tensorflow_gpuenv tensorflow-gpu
conda activate tensorflow_gpuenv

注意:务必注意一点,在安装完tensroflow后,由于我们是新创建的conda环境,该环境中基本上是空的,有很多包和IDE并没有安装进来,例如“Ipython”,“spyder”此时如果我们在该环境下打开spyder/Ipyton/jupyter notebook等,会发现其实IDE使用的kernel并不是新建立的这个环境的kernel,而是“base”这个环境的,而“base”环境中我们并没有安装tensorflow,所以一定无法import。这也就是为什么有很多人在安装好tensorflow后仍然在IDE里无法正常使用的原因了。

通过以下命令安装Anaconda基础包

conda install anaconda

3. 测试

1.查看是否使用GPU

import tensorflow as tf
tf.test.gpu_device_name()

img

2.查看在使用哪个GPU

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

img

  1. 在命令行中,进入python,并输入以下代码:

    import tensorflow as tf
    hello = tf.constant('Hello, TensorFlow!')
    sess = tf.Session()
    print(sess.run(hello))
    # 输出:
    b’ Hello, TensorFlow!
    

附:

参考网址:https://www.anaconda.com/tensorflow-in-anaconda/

This blog is under a CC BY-NC-SA 3.0 Unported License
本文链接:https://idforhyit.github.io/2019/03/10/Anaconda-install-TensorFlow/