AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-15 20:45:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update info about gpu acceleration issue.

Browse Source
This commit is contained in:
catscarlet 2025-02-28 17:08:34 +08:00
parent aa6fb76e0f
commit adfe3a06c6
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -100,14 +100,14 @@ pip install "rembg[cli]" # for library + cli
Otherwise, install `rembg` with explicit CPU/GPU support. Otherwise, install `rembg` with explicit CPU/GPU support.
CPU support: ### CPU support:
```bash ```bash
pip install rembg[cpu] # for library pip install rembg[cpu] # for library
pip install "rembg[cpu,cli]" # for library + cli pip install "rembg[cpu,cli]" # for library + cli
``` ```
GPU support: ### GPU support:
First of all, you need to check if your system supports the `onnxruntime-gpu`. First of all, you need to check if your system supports the `onnxruntime-gpu`.
@ -124,6 +124,8 @@ pip install "rembg[gpu]" # for library
pip install "rembg[gpu,cli]" # for library + cli pip install "rembg[gpu,cli]" # for library + cli
``` ```
Nvidia GPU may require onnxruntime-gpu, cuda, and cudnn-devel. [#668](https://github.com/danielgatis/rembg/issues/668#issuecomment-2689830314) . If rembg[gpu] couldn't work probably and your can't install cuda or cudnn-devel, use rembg[cpu] and onnxruntime instead.
## Usage as a cli ## Usage as a cli
After the installation step you can use rembg just typing `rembg` in your terminal window. After the installation step you can use rembg just typing `rembg` in your terminal window.