* Addressing review comment: Revert the changes to TOML file

Signed-off-by: Pradnya Khalate <[email protected]>
NVIDIA · Feb 25, 2025 · bd919a8 · bd919a8
1 parent acf05c2
commit bd919a8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,10 +22,10 @@ dependencies = [
   'numpy >= 1.24',
   'scipy >= 1.10.1',
   'requests >= 2.31',
-  'nvidia-cublas-cu12 ~= 12.8',
-  'nvidia-cuda-runtime-cu12 ~= 12.8',
-  'nvidia-cusolver-cu12 ~= 11.7',
-  'nvidia-cuda-nvrtc-cu12 ~= 12.8'
+  'nvidia-cublas-cu12 ~= 12.0',
+  'nvidia-cuda-runtime-cu12 ~= 12.0',
+  'nvidia-cusolver-cu12 ~= 11.4',
+  'nvidia-cuda-nvrtc-cu12 ~= 12.0'
 ]
 classifiers = [
     'Intended Audience :: Science/Research',