diff --git a/src/main/python/systemds/__init__.py b/src/main/python/systemds/__init__.py index a618ff6e9dd..d015763a1ec 100644 --- a/src/main/python/systemds/__init__.py +++ b/src/main/python/systemds/__init__.py @@ -26,8 +26,8 @@ __all__ = ["context", "operator", "examples"] required_packages = [ - ("torch", "2.5.1"), - ("torchvision", "0.20.1"), + ("torch", "2.4.1"), + ("torchvision", "0.19.1"), ("librosa", "0.10.2"), ("opencv-python", "4.10.0.84"), ("opt-einsum", "3.3.0"),