1.首先我们先安装pip以及virtualenv
$ sudo apt-get install python-pip python-dev python-virtualenv # 安装python2.x
$ sudo apt-get install python3-pip python3-dev python-virtualenv # 安装python3.x
2.更新系统的pip3源:
编辑 pip 配置文件,将 index-url 修改为 https://mirrors.ustc.edu.cn/pypi/web/simple 。
pip 的配置文件一般位于$HOME/.config/pip/pip.conf(如果没有,请直接创建):
pip.conf 文件配置示例如下:
[global]
index-url = https://mirrors.ustc.edu.cn/pypi/web/simple
format = columns
3.创建 Virtualenv 环境
$ virtualenv --system-site-packages tensorflow # for Python 2.7
$ virtualenv --system-site-packages -p python3 tensorflow # for Python 3.n
4.激活 Virtualenv 环境
$ source ~/tensorflow/bin/activate # bash, sh, ksh, or zsh
$ source ~/tensorflow/bin/activate.csh # csh or tcsh
$ . ~/tensorflow/bin/activate.fish # fish
5.安装tensorflow
(tensorflow)$ pip install --upgrade tensorflow-gpu==1.14.0 # for Python 2.7
(tensorflow)$ pip3 install --upgrade tensorflow-gpu==2.0.0 # for Python 3.n
6.测试tensorflow
$ python
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print sess.run(a+b)
42
>>>
7.退出VirtualEnv
# 当使用完 TensorFlow
(tensorflow)$ deactivate # 停用 virtualenv
$ # 你的命令提示符会恢复原样