基于Anaconda3安装tensorflow-gpu=1.8.0和keras-gpu=2.1.6

基于Anaconda3安装tensorflow-gpu=1.8.0和keras-gpu=2.1.6

1.安装Anaconda

可以在清华镜像站或者anaconda官网下载anaconda3-5.2.0-Linux.sh,不要下载5.3版本,5.3版本使用Python3.7,不能使用tensorflow。

下载完成后,在下载文件所在目录输入以下命令:

$bash Anaconda3-5.2.0-Linux-x86_64.sh

在安装过程中会询问是否添加环境变量,最好选同意,不然需要后续自行安装。

2.安装tensorflow-gpu

Anaconda自带的包安装命令conda可以安装tensorflow和keras,相当于linux中的pip安装命令。

首先要建立一个tensorflow-gpu环境:

$conda create -n tensorflow-gpu python=3.6

然后激活这个环境:

$source activate tensorflow-gpu

就会进入到(tensorflow-gpu)环境中。接着输入安装tensorflow的命令:

$conda install --channel http://conda.anaconda.org/anaconda tensorflow-gpu=1.8.0

安装完成。
输入python对tensorflow进行测试:

$python
>>>import tensorflow as tf

如果输出没有问题,就代表安装正确。

3.安装keras

$conda install keras-gpu=2.1.6

输入python对keras进行测试:

$python
>>>import keras

如果输出有:

using TensorFlow backend.

就代表安装正确。

上述测试程序需要在激活tensorflow-gpu环境的条件下进行。

你可能感兴趣的:(基于Anaconda3安装tensorflow-gpu=1.8.0和keras-gpu=2.1.6)