diff --git a/README.md b/README.md index b32482f..b57402f 100644 --- a/README.md +++ b/README.md @@ -78,7 +78,7 @@ python run.py --file image.jpg --model mld-tresnetd.6-30000 ## Using GPU -Requires CUDA 12.2 and cuDNN9.x. +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/ @@ -89,4 +89,4 @@ https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requ ## Copyright -Public domain, except borrowed parts (e.g. `dbimutils.py`) \ No newline at end of file +Public domain, except borrowed parts (e.g. `dbimutils.py`)