docker ubuntu 16.04 Machine Learning 環境架設

Machine Learning 環境架設

首先先去 docker hub 找到 ubuntu 官方 repository
https://hub.docker.com/_/ubuntu/

docker pull ubuntu
docker run ubuntu:16.04
docker ps -a

看到這個

CONTAINER ID        IMAGE               COMMAND                  CREATED             STATUS                      PORTS               NAMES
d163eaf5caa3        ubuntu:16.04        "/bin/bash"              6 seconds ago       Exited (0) 5 seconds ago                        brave_poincare

因為沒有指定 NAME 所以系統隨機給了一個名字 brave_poincare

啟動 container

docker start brave_poincare

看到名字被顯示出來
brave_poincare

接著進入 container

docker exec -t -i brave_poincare /bin/bash

先 update 再安裝 git, vim

apt-get install update
apt-get install git
apt-get install vim

安裝 python3

apt-get install python3

安裝 pip3

apt-get install python3-pip
apt-get install python3-tk
pip3 install --upgrade pip

安裝常用的套件 numpy, pandas, scipy, matplotlib 等

pip3 install numpy scipy matplotlib ipython jupyter pandas sympy nose

安裝 Deep Learning 相關套件

pip3 install tensorflow
pip3 install keras
pip3 install tqdm

保存 model 相關套件

pip3 install h5py
apt-get install libhdf5-dev -y
apt-get install python3-h5py -y
pip3 install Pillow

安裝 opencv

# cv2 相關
apt-get install build-essential cmake git pkg-config -y
apt-get install libjpeg8-dev libtiff5-dev libjasper-dev libpng12-dev -y
apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev -y
apt-get install libgtk2.0-dev -y
apt-get install libatlas-base-dev gfortran -y
pip3 install opencv-python

打包上傳

commit
(taochen1114 是我在 dockerhub 上的帳號) 

docker commit (我的containerID) taochen1114:tao-ml-ubuntu

修改 tag

docker tag (我的imageID) taochen1114/tao-ml-ubuntu:v2.0

login

docker login

上傳至 docker hub

docker push taochen1114/tao-ml-ubuntu:v2.0

可以在這邊找到
https://hub.docker.com/r/taochen1114/tao-ml-ubuntu/

使用

掛載資料夾

要注意的是，-v 選項的後面是先放「實體的完整路徑」
-v 電腦本機的路徑:docker container中的路徑

docker run --name tao-ml -v /Users/tao/Desktop/pytorch:/home/tao/pytorch -it taochen1114/tao-ml-ubuntu /bin/bash

加上 port mapping (這樣才好使用 jypyter notebook)

docker run --name tao-ml -v /Users/tao/Desktop/pytorch:/home/tao/pytorch -p 8888:8888 -it taochen1114/tao-ml-ubuntu /bin/bash

在 container 內部使用 jupyter

jupyter notebook --no-browser --ip=0.0.0.0 --allow-root --NotebookApp.token=""

進階

安裝 pytorch 套件

# 在 container 裡面做這件事
pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
pip install torchvision