diff --git a/setup.py b/setup.py
index 8b0702f9..ce9a2956 100644
--- 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",