基于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环境的条件下进行。