如何使用docker一键配置深度学习环境并运行程序

首先要安装docker

此处引用官方文档：https://docs.docker.com/install/linux/docker-ce/ubuntu/

 sudo apt-get remove docker docker-engine docker.io containerd runc
sudo apt-get update
sudo apt-get install \
    apt-transport-https \
    ca-certificates \
    curl \
    gnupg-agent \
    software-properties-common

 curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo apt-key fingerprint 0EBFCD88
    
pub   rsa4096 2017-02-22 [SCEA]
      9DC8 5822 9FC7 DD38 854A  E2D8 8D81 803C 0EBF CD88
uid           [ unknown] Docker Release (CE deb) <docker@docker.com>
sub   rsa4096 2017-02-22 [S]

sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"
sudo apt-get update

 sudo apt-get install docker-ce docker-ce-cli containerd.io
sudo docker run hello-world

然后去dockerhub拉一个深度学习docker
sudo docker pull tensorflow/tensorflow:latest-gpu-py3
进入docker中的bash，并安装自己需要的包
sudo docker run -t -i tensorflow/tensorflow:latest-gpu-py3 /bin/bash

安装好之后保存docker镜像
先使用 docker ps 查找当前镜像
在使用docker commit 保存当前快照
docker commit -p 2ee1554a50de ubuntu-backup（ubuntu-backup这个名称随便编

现在环境就搭好了，如何运行程序呢
sudo docker run -p 5000:5000 -it  --ipc=host -v /home/bing/WorkSpace/DeepPink++:/workspace  tensorflow/tensorflow:latest-gpu-py3 /bin/bash

红字表示将此目录放在docker中的workspace目录下
 
posted @ 2019-12-26 21:49  shensobaolibin  阅读(667)  评论(0)  编辑  收藏  举报