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）

 deb 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 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     

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

原文链接：https://blog.csdn.net/smallEngineer/article/details/125817770