1. 程式人生 > >ubuntu上安裝keras + tensorflow

ubuntu上安裝keras + tensorflow

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()