使用 Virtualenv 进行安装TensorFlow
环境:ubuntu 16.04
安装 pip 和 Virtualenv
1
2$ sudo apt-get install python-pip python-dev python-virtualenv # for Python 2.7
$ sudo apt-get install python3-pip python3-dev python-virtualenv # for Python 3.n创建 Virtualenv 环境
1
2$ virtualenv --system-site-packages targetDirectory # for Python 2.7
$ virtualenv --system-site-packages -p python3 targetDirectory # for Python 3.ntargetDirectory
用于指定 Virtualenv 目录。我们假定创建的虚拟环境为tensorflow
,即targetDirectory
为tensorflow
。激活 Virtualenv 环境
1
$ source ~/tensorflow/bin/activate
执行上述
source
命令后,您的提示符应该会变成如下内容:1
(tensorflow)$
安装或更新
pip
1
(tensorflow)$ easy_install -U pip
安装
TensorFlow
1
2
3
4(tensorflow)$ pip install --upgrade tensorflow # for Python 2.7
(tensorflow)$ pip3 install --upgrade tensorflow # for Python 3.n
(tensorflow)$ pip install --upgrade tensorflow-gpu # for Python 2.7 and GPU
(tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU备注:如果安装失败或者网速慢,可参考TensorFlow-CPU/GPU安装ubuntu16.04版 ,离线安装
tensorflow
。退出 Virtualenv 环境
1
(tensorflow)$ deactivate
验证环境
激活 Virtualenv 环境
1
$ source ~/tensorflow/bin/activate
1
2
3
4
在 Virtualenv 环境激活后,您就可以从这个 shell 运行 TensorFlow 程序。您的提示符将变成如下所示,这表示您的 Tensorflow 环境已处于活动状态:
-(tensorflow)$
1
2
输入代码验证import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
print(sess.run(hello))
1
2
3
4
- 看到如下输出,表示安装正确
-Hello, TensorFlow!
1
2
用完 TensorFlow 后,可以通过发出以下命令调用 `deactivate` 函数来停用环境:(tensorflow)$ deactivate
1
2
3
4
- 提示符将恢复为您的默认提示符 ,就表示退出了。
-$
1
2
3
4
5
6
7
### 卸载 Virtualenv 环境 `targetDirectory`
要卸载你刚刚新建的Virtualenv 环境 `TensorFlow`的话,只需要删除该文件夹就可以了。
$ rm -r targetDirectory # 在这里 targetDirectory改为你创建的环境名字`
转载请注明:Seven的博客