diff --git a/src/otx/cli/utils/installation.py b/src/otx/cli/utils/installation.py index 0a554c3a3b3..532bb487400 100644 --- a/src/otx/cli/utils/installation.py +++ b/src/otx/cli/utils/installation.py @@ -32,6 +32,7 @@ "2.3.0": {"torchvision": "0.18.0", "cuda": ("11.8", "12.1")}, "2.3.1": {"torchvision": "0.18.1", "cuda": ("11.8", "12.1")}, "2.4.0": {"torchvision": "0.19.0", "cuda": ("11.8", "12.1", "12.4")}, + "2.5.1": {"torchvision": "0.20.1", "cuda": ("11.8", "12.1", "12.4")}, } MM_REQUIREMENTS = [