欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu上安装keras + tensorflow

程序员文章站 2022-05-26 19:01:39
...

keras最近很火,它是一个较高层次的深度学习框架,后端可以使用tensorflow或者theano。尝试着在ubuntu上面安装了keras和tensorflow,过程不算麻烦,但是还是有些小问题需要注意的。

1. 安装tensorflow。

tensorflow有很多种安装方式,参见网址:

http://www.tensorfly.cn/tfdoc/get_started/os_setup.html

最简单的方式是二进制安装,以上网址中提供的二进制安装命令为:

pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
按照上面的命令可以把tensorflow安装到默认目录/usr/local/下。需要注意的时,上面命令安装的是0.5.0版的,稍有些旧。可以用下面的命令安装1.0.0版的:

pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.0-cp27-none-linux_x86_64.whl

2. 安装protobuf。

tensorflow需要protobuf工具的支持,该工具的安装被分为两个步骤,分别为安装protobuf-cpp-3.0.0和protobuf-python-3.0.0(这里用的是3.0版,其他版本应该也可以)。前者安装.so类型的动态库和可执行文件protoc,后者为python提供支持,都不可或缺。

2.1 安装protobuf-cpp-3.0.0。

下载protobuf-cpp-3.0.0.tar.gz

tar -zxvf protobuf-cpp-3.0.0.tar.gz

cd protobuf-cpp-3.0.0

./configure

make

make install

上述命令把protobuf安装到默认目录/usr/local/下面。

可以通过以下命令测试protobuf是否安装成功

protoc --version

2.2 安装protobuf-python-3.0.0。

下载protobuf-python-3.0.0.tar.gz

tar -zxvf protobuf-python-3.0.0.tar.gz

cd protobuf-python-3.0.0/python

python setup.py build

python setup.py install

把protobuf对python的支持安装到默认目录/usr/local下面,具体为

/usr/local/lib/python2.7/dist-packages/protobuf-3.0.0-py2.7.egg

为了在python下面能用,还要执行下述命令

cd /usr/local/lib/python2.7/dist-packages

ln -s ./protobuf-3.0.0-py2.7.egg/google google

也就是在目录

/usr/local/lib/python2.7/dist-packages

下面做一个指向

/usr/local/lib/python2.7/dist-packages/protobuf-3.0.0-py2.7.egg/google

的软链接。

可以通过以下命令测试安装python protobuf是否成功。

python

>>> import google.protobuf

3. 安装keras

最简单的方法还是安装二进制文件。命令如下

sudo pip install keras
然后可以通过简单的命令来判断keras是否安装成功。

python
>>>from keras.models import Sequential
>>>model = Sequential()