用docker借助deepo镜像训练深度学习模型

用docker借助deepo镜像训练深度学习模型

Deepo简介

deepo是我从网上了解的一个比较全的深度学习环境镜像，有助于在docker中快速配置深度学习研究环境，支持几乎所有常用的深度学习框架

docker hub 地址

https://hub.docker.com/r/ufoym/deepo

安装步骤

1. 此处安装的为显卡版本，参考官网的步骤，首先要安装安装Docker和nvidia-docker

   Docker 之前就已经装好，不再赘述，安装nvidia-docker让docker容器中能调用显卡训练

   参考博客：https://blog.csdn.net/Victor2code/article/details/106226622?spm=1001.2014.3001.5506

   和官网：https://nvidia.github.io/nvidia-docker/

   来安装nvidia-docker2

   安装完成后，运行

   docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
   

   
   无误后继续

2. 从Docker Hub获取一体化镜像

   docker pull ufoym/deepo
   

   运行