diff --git a/README.md b/README.md index 4057293..b32482f 100644 --- a/README.md +++ b/README.md @@ -78,13 +78,13 @@ python run.py --file image.jpg --model mld-tresnetd.6-30000 ## Using GPU -Requires CUDA 12.2 and cuDNN8.x. +Requires CUDA 12.2 and cuDNN9.x. ``` pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ ``` -https://onnxruntime.ai/docs/install/ +https://onnxruntime.ai/docs/install/
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements ## Copyright diff --git a/requirements.txt b/requirements.txt index ac86f53..40be636 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ -pandas==2.2.2 -huggingface-hub==0.22.2 -Pillow==10.0.1 -deepdanbooru==1.0.2 -onnxruntime==1.17.3 -numpy==1.26.4 -opencv-python==4.9.0.80 +pandas==2.2.3 +huggingface-hub==0.28.1 +Pillow==11.1.0 +deepdanbooru==1.0.3 +onnxruntime==1.20.1 +numpy==2.2.2 +opencv-python==4.11.0.86