搜索
您的当前位置:首页正文

TensorFlow 安装小记

来源:二三娱乐

安装系统: Ubuntu 17.04

  • virtualenv
  • "native" pip
  • Docker
  • Anaconda

这里选择virtunalenv,使用这种方法的好处就是在一个隔离的环境中安装,不影响系统已有的python环境。
所以安装会放在一个新创建的目录中,删除该目录就相当于卸载tensorflow了。

命令如下:

sudo apt-get install python-pip python-dev python-virtualenv 
virtualenv --system-site-packages tensorflow
cd ~/tensorflow
source ~/tensorflow/bin/activate
pip install --upgrade tensorflow      # for Python 2.7 just CPU

到这一步会发现下载超慢,因为某些不可抗力的原因。解决方法是pip源使用国内镜像,如:

永久使用的话创建文件 ~/.pip/pip.conf,配置如下:

[global]
index-url = 
[install]

这之后下载就飞快了。

安装好之后,每次使用virtualenv环境都需要激活, 输入命令:

source ~/tensorflow/bin/activate

退出则输入:

deactivate

验证是否安装成功,试着输出: hello world


Screenshot from 2017-05-12 22-01-37.png

测试MNIST数据

  1. 从 上下载4个gz格式文件,放在新创建的目录MNIST_DATA下
  2. 下载运行脚本

然后运行:

python mnist_softmax.py --data_dir ./MNIST_data/

输出如下:

Extracting ./MNIST_data/train-images-idx3-ubyte.gz
Extracting ./MNIST_data/train-labels-idx1-ubyte.gz
Extracting ./MNIST_data/t10k-images-idx3-ubyte.gz
Extracting ./MNIST_data/t10k-labels-idx1-ubyte.gz
0.9194
Top