Add cpu only option instead of gpu option.
This commit is contained in:
@@ -24,7 +24,7 @@ options:
|
||||
Prediction threshold (default is 0.35)
|
||||
--ext EXT Extension to add to caption file in case of dir option (default is .txt)
|
||||
--overwrite Overwrite the file if it exists
|
||||
--gpu Use GPU (Compatible CUDA and cuDNN required)
|
||||
--cpu Use CPU only
|
||||
--model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,mld-caformer.dec-5-97527,mld-tresnetd.6-30000}
|
||||
modelname to use for prediction (default is wd14-convnextv2.v1)
|
||||
```
|
||||
@@ -63,9 +63,5 @@ Requires CUDA 12.2 and cuDNN8.x.
|
||||
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
|
||||
```
|
||||
|
||||
```
|
||||
python run.py --file image.jpg --gpu
|
||||
```
|
||||
|
||||
https://onnxruntime.ai/docs/install/
|
||||
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements
|
||||
|
||||
Reference in New Issue
Block a user