在服务器Ubuntu22.04系统下,IOPaint的部署
一、安装环境:ubuntu基本环境配置
1.更新包列表:
修改配置文件
vi /etc/apt/sources.list
将原有的内容注释(每行最前面加#)
在最后添加以下内容
######## Ubuntu22.04LTS 清华镜像源 ###############
# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-security main restricted universe multiverse
# 预发布软件源,不建议启用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-proposed main restricted universe multiverse
更新软件列表
sudo apt-get update
更新本地软件
sudo apt upgrade
更新时间较长,请耐心等待
2. 安装GPU驱动程序
- 这个命令会列出您的系统硬件信息,以及推荐的驱动程序版本
ubuntu-drivers devices
- 这里的4090建议安装 nvidia-driver-535 驱动程序
sudo apt update
sudo apt install nvidia-driver-535
- 安装完成后,请务必重启您的系统,以便新的驱动程序生效。
reboot
3.安装 NVIDIA Container Toolkit 软件包:
sudo apt install nvidia-container-runtime
apt install nvidia-cuda-toolkit
4.查看显卡信息
nvidia-smi
二、安装miniconda环境
1. 下载miniconda3
wget https://mirrors.cqupt.edu.cn/anaconda/miniconda/Miniconda3-py310_23.10.0-1-Linux-x86_64.sh
2. 安装miniconda3
bash Miniconda3-py310_23.10.0-1-Linux-x86_64.sh -u
直接一直enter键,到输入路径和yes
这边建议路径为:miniconda3
3. 打开用户环境编辑页面
vim ~/.bashrc
- 点击键盘I键进入编辑模式,在最下方输入以下代码
export PATH="/root/miniconda3/bin:$PATH"
4. 重新加载用户环境变量
source ~/.bashrc
5. 初始化conda
conda init bash
source ~/.bashrc
6. 初始化conda
conda -V
7. 创建IOPaint需要的环境
conda create -n iopaint python=3.9
conda activate iopaint
三、克隆IOPaint仓库
官网地址:IOPaintGitHub地址
git clone https://github.com/Sanster/IOPaint.git
如果下载速度较慢可添加https://mirror.ghproxy.com/
镜像源
git clone https://mirror.ghproxy.com/https://github.com/Sanster/IOPaint.git
进入到 IOPaint 文件夹内
cd IOPaint
pip3 install iopaint
iopaint start --model=lama --device=cuda --port=8080 --host 0.0.0.0
下载较慢可以在前面加个镜像HF_ENDPOINT=https://hf-mirror.com
HF_ENDPOINT=https://hf-mirror.com iopaint start --model=lama --device=cuda --port=8080 --host 0.0.0.0
--port 8080: 设置服务器监听的端口为业务端口,此处改为业务端口。
--model=lama:指定了以lama模型(图像消除)启动IOPaint
--device=cuda:指定了应用程序应该使用 CUDA 兼容的 GPU 设备来运行
--host 0.0.0.0:用于指定应用程序监听的 IP 地址
启动成功后,访问http://主机IP:端口