背景:服务器的cudnn版本太低了,没有权限去修改。故新建包含cuda和cudnn的docker
步骤
一、拉取镜像及创建docker
- 拉取相关的镜像
-
从镜像列表选出相关版本的镜像https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/supported-tags.md
-
在ubuntu系统中拉取镜像
docker pull nvidia/cuda:12.0.1-cudnn8-devel-ubuntu20.04
(按照官方docker pull nvidia/cuda
拉取最近的镜像会报错)
-
- 根据镜像创建docker
docker run -itd --gpus gpu数量 \
--name=新建docker名 \
--shm-size=指定共享内存大小 \
-v 服务器地址:映射地址 镜像名
docker run -itd --gpus 2 \
--name="cuda_1201" \
--shm-size=4g \
-v /data/cv:/mount nvidia/cuda:12.0.1-cudnn8-devel-ubuntu20.04
二、docker内配置网络
- apt install 无法安装包
- 备份docker中的
source.lst
- 在docker中执行
cp /etc/apt/sources.list /etc/apt/sources.list.bk
- 从清华源下载配置文件
- 备份docker中的
- 根据上述的选项得到的文本,复制到
/etc/apt/sources.list
中
- 在dockers中执行
apt update
和apt upgrade
进行完上述步骤后,就可以用apt进行安装包了apt install vim
,apt install python3
,apt install pip3
参考链接
- CUDA 安装教程
- Error response from daemon: manifest for nvidia/cuda:latest not found: manifest unknown: manifest