iopaint是一个开源的AI去水印的工具,可以本地部署

Install pytorch
If you want to run model on GPU, it is necessary to install the GPU version of PyTorch. Otherwise, you can skip this step.

For NVIDIA GPU users

pip3 install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu118

For AMD GPU users, only works on linux, as pytorch is not yet supported on Windows with ROCm.

pip3 install torch==2.1.2 torchvision==0.16.2 --index-url https://download.pytorch.org/whl/rocm5.6

Install IOPaint
pip3 install iopaint

**Start IOPaint server**

iopaint start --model=lama --device=cpu --port=8080

IOPaint is now running at http://localhost:8080(opens in a new tab), you can open it in your browser to start using it.

You can see all command line arguments using iopaint start --help, and you can see more models and features in the models page.


**解决安装Failed building wheel for pillow**
解决error: command ‘x86_64-linux-gnu-gcc‘ failed with exit status 1

apt-get install libjpeg-dev zlib1g-dev

查看自己python的版本,然后下载自己版本Python的devel,比如python3.7.7就是

sudo apt-get install python3.7-dev

sudo apt-get install build-essential python3-dev libssl-dev libffi-dev libxml2 libxml2-dev libxslt1-dev zlib1g-dev

标签: none

添加新评论