使用docker配置深度学习环境

2018.10.20 测试成功 配置是ubuntu16.04+gtx750ti 已经装好了nvidia的驱动 刘泽豪

0.如果之前安装过Docker的旧版本,可以使用下2018.10.20 测试成功 配置是ubuntu16.04+gtx750ti 已经装好了nvidia的驱动 刘泽豪

0.如果之前安装过Docker的旧版本,可以使用下面的命令进行卸载:
sudo apt-get remove docker docker-engine docker.io
1.安装Docker-ce
Step 0:添加阿里Docker镜像源
step 1: 安装必要的一些系统工具
sudo apt-get update
sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common
step 2: 安装GPG证书
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
step 3: 写入软件源信息
sudo add-apt-repository “deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable”
step 4: 更新并安装 Docker-CE
sudo apt-get -y update
sudo apt-get -y install docker-ce

Step 1: 安装必要的一些系统工具
sudo apt-get update sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common

Step 2:安装GPG证书
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -

Step 3: 写入软件源信息
sudo add-apt-repository “deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable”

Step 4:
更新并安装 Docker-CE sudo apt-get -y update sudo apt-get -y install docker-ce

Step 5:验证安装成功
输入 docker version 可以出现版本信息

安装Nvidia-Docker需要18.06以上版本,如果安装低于这个版本,用
sudo apt-get remove docker docker-engine docker.io
卸载低版本

curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun
这个命令重新安装即可

2.安装Nvidia-Docker
Step 0:之前有安装过Nvidia-Docker 1.0版本的,先卸载掉
docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

Step 1:添加软件包库(速度慢的话需要翻墙)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey |
sudo apt-key add -
distribution=$(. /etc/os-release;echo I D ID IDVERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list |
sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

Step 3:安装nvidia-docker2
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

Step 4: 从Docker 中国官方镜像上pull一个cuda镜像试试看
docker pull registry.docker-cn.com/nvidia/cuda

Step 5:验证
docker images 列出所有的镜像
docker run -t -i nvidia/cuda nvidia/cuda是第四步安装的
nvcc -v 出现Cuda信息即为安装成功
ctrl+D 退出界面

3.安装deepo深度学习镜像
docker pull ufoym/deepo 9g左右,安装时间会比较长
docker run -t -i ufoym/deepo 运行deepo
验证:
$ caffe --version caffe环境
$ th pytorch环境
面的命令进行卸载:
sudo apt-get remove docker docker-engine docker.io
1.安装Docker-ce
Step 0:添加阿里Docker镜像源
step 1: 安装必要的一些系统工具
sudo apt-get update
sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common
step 2: 安装GPG证书
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
step 3: 写入软件源信息
sudo add-apt-repository “deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable”
step 4: 更新并安装 Docker-CE
sudo apt-get -y update
sudo apt-get -y install docker-ce

Step 1: 安装必要的一些系统工具
sudo apt-get update sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common

Step 2:安装GPG证书
curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -

Step 3: 写入软件源信息
sudo add-apt-repository “deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable”

Step 4:
更新并安装 Docker-CE sudo apt-get -y update sudo apt-get -y install docker-ce

Step 5:验证安装成功
输入 docker version 可以出现版本信息

安装Nvidia-Docker需要18.06以上版本,如果安装低于这个版本,用
sudo apt-get remove docker docker-engine docker.io
卸载低版本

curl -fsSL https://get.docker.com | bash -s docker --mirror Aliyun
这个命令重新安装即可

2.安装Nvidia-Docker
Step 0:之前有安装过Nvidia-Docker 1.0版本的,先卸载掉
docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker

Step 1:添加软件包库(速度慢的话需要翻墙)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey |
sudo apt-key add -
distribution=$(. /etc/os-release;echo I D ID IDVERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list |
sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

Step 3:安装nvidia-docker2
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

Step 4: 从Docker 中国官方镜像上pull一个cuda镜像试试看
docker pull registry.docker-cn.com/nvidia/cuda

Step 5:验证
docker images 列出所有的镜像
docker run -t -i nvidia/cuda nvidia/cuda是第四步安装的
nvcc -v 出现Cuda信息即为安装成功
ctrl+D 退出界面

3.安装deepo深度学习镜像
docker pull ufoym/deepo 9g左右,安装时间会比较长
docker run -t -i ufoym/deepo 运行deepo
验证:
$ caffe --version caffe环境
$ th pytorch环境

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要在Docker配置深度学习环境,您可以按照以下步骤进行操作: 1. 安装Docker:如果您还没有安装Docker,请根据您的操作系统(Windows、Mac或Linux)在Docker官方网站上下载并安装Docker。 2. 创建一个Docker镜像:您可以使用已经预先配置好的深度学习镜像,例如NVIDIA提供的nvidia-docker镜像,或者您可以自己构建一个包含必要组件和库的镜像。在这里,我们以自己构建镜像为例。 3. 编写Dockerfile:创建一个名为Dockerfile的文本文件,在其中定义您的镜像。以下是一个示例Dockerfile: ``` FROM nvidia/cuda:11.2.0-cudnn8-runtime-ubuntu20.04 # 安装系统依赖 RUN apt-get update && apt-get install -y \ python3 \ python3-pip \ git \ && rm -rf /var/lib/apt/lists/* # 安装深度学习相关库 RUN pip3 install tensorflow-gpu==2.4.1 \ keras==2.4.3 \ numpy \ matplotlib \ jupyterlab # 设置工作目录 WORKDIR /app # 复制项目代码到容器中 COPY . /app # 设置启动命令 CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=8888", "--allow-root"] ``` 在上述示例中,我们使用了基于Ubuntu 20.04的NVIDIA CUDA和cuDNN的运行时镜像,并安装了Python、Git以及一些常用的深度学习库。您可以根据自己的需求自定义Dockerfile。 4. 构建镜像:在创建好Dockerfile后,使用以下命令在终端中构建镜像(确保当前目录包含Dockerfile): ``` docker build -t deep-learning-env . ``` 这将会根据Dockerfile创建一个名为deep-learning-env的镜像。 5. 运行容器:运行以下命令启动一个容器,并将本地的一个目录挂载到容器中,以便在容器中进行文件的读写操作: ``` docker run -it --gpus all -v /path/to/local/directory:/app deep-learning-env ``` 在上述命令中,`/path/to/local/directory`是您本地文件系统中的目录,您可以将其替换为您自己的目录。这样,您就可以在容器中进行深度学习相关的开发和实验了。 希望以上步骤能帮助您成功配置深度学习环境。如有任何问题,请随时向我提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值