NVIDIA-Docker安裝後下載下傳機器學習模型架構

鏡像的基本操作

1.NVIDIA Docker 安裝

1.1 安裝準備

* 安裝CUDA
* 系統設定-->軟體與更新-->下載下傳選擇其他-->在彈框中選擇中國-->選擇mirrors.aliyun.com-->右下角選擇伺服器
* System Settings-->Software &Updates-->Download from-->Other..-->China-mirrors.aliyun.com-->Choose Server-->Close-Reload
           

1.2 docker CE安裝

//更新
sudo apt-get update
sudo apt-get upgrade

//腳本安裝

curl -fsSL get.docker.com -o get-docker.sh

sudo sh get-docker.sh --mirror Aliyun
//當以下指令顯示含有hello world字樣時說明DOCKER環境安裝

sudo docker run hello-world
           

1.3 下載下傳 NVIDIA-docker

wget http://dink.51ai.pro/nvidia-docker_1.0.1-1_amd64.deb

//如果上述下載下傳失敗，下載下傳以下

wget https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb

//安裝
sudo dpkg -i nvidia-docker_1.0.1-1_amd64.deb

           

1.4 檢查nvidia-docker服務是否存在

systemctl list-units --type=service | grep -i nvidia-docker

//如果上述操作失敗，運作以下

systemctl list-units --type=service | grep -i nvidia-docker-plugin

//安裝modprobe

sudo apt-get install nvidia-modprobe
           

2.下載下傳并運作鏡像

2.1 如果鏡像過大,可以設定阿裡雲鏡像加速器

sudo mkdir -p /etc/docker

sudo tee /etc/docker/daemon.json <<-'EOF'
{
  "registry-mirrors": ["https://9pbu6dtx.mirror.aliyuncs.com"]
}
EOF

sudo systemctl daemon-reload

sudo systemctl restart docker
           

2.2 阿裡雲上拉去鏡像

// 例如拉去之前做的機器學習模型訓練鏡像 大概19G
sudo docker pull registry.cn-hangzhou.aliyuncs.com/dink_framework/dink0.22:latest
           

2.3 驗證鏡像：

sudo docker images
           

2.4運作鏡像,生成可視化容器操作:

sh run.sh