2
0
Эх сурвалжийг харах

Update info about gpu acceleration issue.

catscarlet 6 сар өмнө
parent
commit
adfe3a06c6
1 өөрчлөгдсөн 4 нэмэгдсэн , 2 устгасан
  1. 4 2
      README.md

+ 4 - 2
README.md

@@ -100,14 +100,14 @@ pip install "rembg[cli]" # for library + cli
 
 Otherwise, install `rembg` with explicit CPU/GPU support.
 
-CPU support:
+### CPU support:
 
 ```bash
 pip install rembg[cpu] # for library
 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`.
 
@@ -124,6 +124,8 @@ pip install "rembg[gpu]" # for library
 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
 
 After the installation step you can use rembg just typing `rembg` in your terminal window.