Ubuntu系统 nvidia-docker创建深度学习容器

任务：创建新的container，镜像使用ubuntu系统

注意使用nvidia-docker 命令才能使得容器内可以使用GPU资源

Step 1 : 拉取ubuntu镜像

官网寻找需要的镜像：https://hub.docker.com/r/nvidia/cuda/tags?page=1

docker pull image_you_need # 拉取镜像

Step 2 : 创建容器

1、查看现有的容器和镜像

''' 查看所有container '''
docker ps -a

''' 查看所有镜像环境 '''
docker images

2、创建新的容器

nvidia-docker run -tid --name baiyu -p 6002:22 image_you_need 

# 端口映射到: 6002
# 创建容器名: baiyu
# 使用的镜像: image_you_need 

进入该容器：

docker exec -it baiyu /bin/bash

查看是否搞定：

nvidia-smi # 显卡
nvcc -V # cuda版本

 

Step 3 : 其他配置

 

# 首先更新源
apt-get update

# 失败的话参考：https://blog.csdn.net/baiyuwujia/article/details/105100358

 

#安装常用工具
apt-get install -y vim
apt-get install -y tree
apt-get install -y net-tools
#安装pip
apt-get install -y python-pip python-dev build-essential
#安装supervisor
apt-get install -y supervisor

  配置该容器ssh服务

apt-get install openssh-server -y # 安装ssh服务

# 修改配置文件：
vim /etc/ssh/sshd_config

PermitRootLogin yes  
UsePAM no

# 启动 
service ssh start

# 给个密码，可以远程ssh连接root了
passwd root