首先安装Anaconda,下载Anaconda3-5.2.0-Linux-x86_64.sh
运行
sh Anaconda3-5.2.0-Linux-x86_64.sh -b
配置环境变量,修改~/.bashrc,添加Anaconda的环境变量
ANACONDA_HOME=/home/用户名/anaconda3
export PATH=$ANACONDA_HOME/bin:$PATH
直接创建虚拟环境来安装tensorflow2.1,直接安装的话,总有包无法自动卸载导致安装失败。创建虚拟环境
conda create -n tf2 python=3.6 -y
进入虚拟环境
source activate tf2
更新以下pip,否则默认安装tensorflow2.1
pip install --upgrade pip -i https://pypi.douban.com/simple
这里-i,是指定下载源,默认下载源的话,就是龟速
安装tensorflow2.1,同样,指定下载源
pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple/
注意,即使指定下载源,速度很快,也可能不能一次安装成功,所以如果出现了time out,那就多运行几次吧,真没辙。
成功安装后,如下图所示
之所以非要在ubuntu下安装tensorflow2.1是因为要试一试albert,不知道普通笔记本上的虚拟机能不能跑的起来。
tensorflow2.1版的进来的警告倒是舒服多了,1.x的上来十几行警告,看着就难受。