Docker部署ubuntu18.04深度学习环境

1. 查看CUDA版本

cat  /usr/local/cuda/version.txt

2. 查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

下图中6为版本

Pull Docker image
1. 拉

docker pull nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04

2. 查

docker images

3. 启动(添加端口映射)

sudo docker run -it --name algorithm -v /data/algorithm:/data/algorithm -p 8080:22 --runtime=nvidia -e NVIDIA_VISIBLE_DEVICE=all nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04

进入镜像

docker exec -it algorithm /bin/bash

不停止 退出镜像

ctrl p + ctrl q

测试显卡驱动

nvidia-smi

问题一
docker中无法用sudo
解决方法

apt-get update
apt-get install sudo

问题二
pip更新为清华源

pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

问题三
安装opencv
首先更换清华源
之后：

apt-get update && apt-get install -y python3-opencv

问题四 安装git

sudo apt-get install git

问题五 安装vim

sudo apt-get install vim-gtk

清华源（Ubuntu 18.04）

# 默认注释了源码镜像以提高 apt update 速度，如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse

# 预发布软件源，不建议启用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse

Pycharm 远程连接 Docker

1. 更新容器的apt源，安装ssh和vim

apt-get update
apt-get install openssh-server
apt-get install vim

2. 使用vim打开并修改配置文件，找到PermitRootLogin prohibit-password这一行，修改为PermitRootLogin yes，允许通过ssh远程访问docker。

vim /etc/ssh/sshd_config

3. 创建docker中root用户的密码。

passwd root

4. 启动ssh服务，至此，服务器端配置完毕。

service ssh restart

vim全部删除：按esc后，然后dG
全选（高亮显示）：按esc后，然后ggvG或者ggVG