天天看点

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

一份Linux下新建tensorflow环境的教程想分享给有需要的人~

安装步骤

1. update下载最新的软件安装包,upgrade安装已下载的最新的安装包

sudo apt update

sudo apt upgrade

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)
Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

2.    添加cuda环境变量

 进入个人目录下,打开 .bashrc 文件,在后面添加cuda路径

export PATH=/usr/local/cuda-8.0/bin/:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

保存退出后, 运行 source .bashrc ,并通过 nvcc -V命令查看是否添加成功。

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

3.    安装Anaconda

在某个用户目录下下载好Anaconda 的安装文件,可以直接复制相应版本的安装文件到自己的用户目录下进行安装。

运行  bash Anaconda2-5.0.1-Linux-x86_64.sh 进行安装

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)
Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)
Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

此处选择yes,自动将Anaconda2添加到 .bashrc 中的PATH变量中。

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

运行source.bashrc 然后输入Python,现在Python运行的已经是Anconda的Python

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

现在Anaconda已经安装好了。因为Anaconda.org的服务器在国外,下载速度比较慢,可以使用清华的镜像:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

4.    新建Anaconda虚拟环境

conda create -n 虚拟环境名字   python=版本号
Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

可以通过 conda info -e 命令查看当前所有的Anaconda虚拟环境。默认当前所在的环境是root环境,通过  source activate 虚拟环境名  来激活对应的虚拟环境。

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

5.    安装TensorFlow-GPU

可以选择将tensorflow-gpu安装到指定的虚拟环境中,此处安装到刚刚新建的 tf13中。通过Anaconda可以安装指定版本的tensorflow,此处安装tensorflow1.3 版本。TensorFlow已经更新到1.4版本,但是通过Anaconda源中最新的tensorflow只有1.3版本的,想要安装tensorflow1.4版本的可以安装官网上教程进行安装。

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)
Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

验证tensorflow是否安装成功

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

6.    安装keras-gpu

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

验证keras是否安装成功。

Linux下新建tensorflow环境(Anaconda+ tensorflow + keras)

至此,Anaconda+ tensorflow + keras 已经安装成功了。