2.安装docker依赖
sudo apt-get install ca-certificates curl gnupg lsb-release
3.添加docker密钥curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -
4.添加阿里云docker软件源
sudo add-apt-repository 'deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable'
5.安装docker
apt-get install docker-ce docker-ce-cli containerd.i
6.docker配置国内镜像源
vi /etc/docker/daemon.json
按i进入编辑模式 加入以下内容:
{
'registry-mirrors': [
'https://docker.mirrors.ustc.edu.cn',
'https://hub-mirror.c.163.com',
'https://docker.m.daocloud.io',
'https://ghcr.io',
'https://mirror.baidubce.com',
'https://docker.nju.edu.cn'
]
}
按ESC键退出编辑模式,接着输入:wq,保存并退出
7.重新加载docker
sudo systemctl daemon-reload
8.重启docker
sudo systemctl restart docker
三、安装英伟达显卡驱动
•配置存储库
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
| sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
| sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
| sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
•更新包列表
•安装 NVIDIA Container Toolkit 软件包sudo apt-get install -y nvidia-container-toolkit
•配置 Docker 以使用 Nvidia 驱动程序sudo nvidia-ctk runtime configure --runtime=docker
•重新启动docker
sudo systemctl restart docker
•从 Docker CodeFormer 项目下载代码,并进入项目目录git clone https://github.com/soulteary/docker-codeformer.git
cd docker-codeformer
•执行项目中的镜像构建工具因为项目锁定了 Python 3.8,所以我们暂时只能使用 nvidia/pytorch:23.04-py3 来作为基础镜像。
•在完成基础镜像构建之后
可以从网盘下载 weights.zip
模型应用运行需要的所有模型都在这里了
下载完毕后,解压缩模型压缩包,将 CodeFormer、facelib、realesrgan 三个目录放置到 weights 目录中
•完整的项目结构如下图
•准备好模型文件之后,使用下面的命令启动模型应用docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -it -v `pwd`/weights/:/app/CodeFormer/weights -p 7860:7860 soulteary/docker-codeformer
稍等片刻,我们将会看到类似 Running on local URL: http://0.0.0.0:7860 To create a public link, set 'share=True' in 'launch()'.的日志
•执行安装命令
将7860改成业务端口
这里是30137 所以:
docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -it -v `pwd`/weights/:/app/CodeFormer/weights -p 30137:7860 soulteary/docker-codeformer
启动成功后,访问http://主机IP:端口