1. 程式人生 > >利用docker安裝keras映象

利用docker安裝keras映象

本身並不複雜的操作,由於自己實在太小白,折騰了好久,現在還是有不明白的地方,不過先安裝上再慢慢學習。

1、製作keras的docker映象
GitHub上keras的原始碼是提供了Dockerfile的,可以用來製作映象,網址:
https://github.com/keras-team/keras/blob/master/docker/Dockerfile
但是由於官網提供的是GPU版本,而我們用的虛擬機器,需要安裝CPU版本的,因此對Dockerfile做修改,修改後(由於貼上整個檔案內容用markdown會出奇怪的問題,所以這裡直接上傳檔案):
修改後的Dockerfile檔案

build命令生成映象:

docker build -t keras_docker /root/keras_dockerfile

keras_docker 是自己定義的生成映象的標籤,後面是修改後dockerfile所存放的目錄

2、run:

docker run -it -p 8886:8888 -v /keras:/notebooks/kerasdata keras_docker:latest

冒號前面是上一步build時所定義的標籤

3、在瀏覽器訪問 伺服器地址:8886
輸入token後的口令
注意這裡和安裝tensorflow映象一樣,也是不要停止執行程式,可以關掉這個連線讓它在後臺執行

4、測試
執行example的方法,進入example資料夾,新建一個python文件(ipynb檔案),然後在cell內輸入%run mnist_cnn.py,並執行,即可成功執行example資料夾中的mnist_cnn.py檔案

由於執行時需要下載資料集,可能出現網路不穩定而報timeouterror的情況,再次嘗試即可成功執行。