Update versions

NKI-AI · Jun 13, 2024 · a8f70d4 · a8f70d4
1 parent bf813b9
commit a8f70d4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -43,6 +43,7 @@ def finalize_options(self):
         "OSI Approved :: Apache Software License",
         "Natural Language :: English",
         "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
     ],
@@ -56,9 +57,9 @@ def finalize_options(self):
     install_requires=[
         "numpy>=1.21.2",
         "h5py==3.11.0",
-        "omegaconf==2.1.1",
-        "torch>=1.10.2",
-        "torchvision",
+        "omegaconf==2.3.0",
+        "torch>=2.0.0",
+        "torchvision==0.17.0",
         "scikit-image>=0.19.0",
         "scikit-learn>=1.0.1",
         "tensorboard>=2.7.0",