diff --git a/README.md b/README.md index dca8bbe..b81e488 100644 --- a/README.md +++ b/README.md @@ -38,15 +38,15 @@ To enable GPU, install the [CUDA-enabled version](https://pytorch.org/get-starte Python Package Dependencies * dill>=0.4.0 * numpy>=2.0.2 -* onnxruntime-gpu>=1.19.2 +* onnxruntime-gpu>=1.27.0 * pandas>=2.2.2, <3.0.0 * pillow>=11.0.0 * pyexiftool>=0.5.6 * opencv-python>=4.12.0.88 * scikit-learn>=1.5.2 * timm>=1.0.9 -* torch>=2.6.0 -* torchvision>=0.21.0 +* torch>=2.11.0 +* torchvision>=0.26.0 * tqdm>=4.66.5 * ultralytics>=8.3.95 * wget>=3.2 diff --git a/pyproject.toml b/pyproject.toml index d45f40b..f97d7b8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,15 +13,15 @@ requires-python = ">=3.12" dependencies = [ "dill>=0.4.0", "numpy>=2.0.2", - "onnxruntime-gpu>=1.19.2", + "onnxruntime-gpu>=1.27.0", "pandas>=2.2.2,<3.0.0", "pillow>=11.0.0", "pyexiftool>=0.5.6", "opencv-python>=4.12.0.88", "scikit-learn>=1.5.2", "timm>=1.0.9", - "torch>=2.6.0", - "torchvision>=0.21.0", + "torch>=2.11.0", + "torchvision>=0.26.0", "tqdm>=4.66.5", "ultralytics>=8.3.95", "wget>=3.2"