如何使用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) <[email protected]> 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目录下
 

猜你喜欢

转载自www.cnblogs.com/shensobaolibin/p/12104634.html
今日推荐