Fix cuda version dep for nvidia-apex

ctuning · Feb 14, 2024 · 2e1e8e9 · 2e1e8e9
1 parent e664c5c
commit 2e1e8e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cm-mlops/script/get-generic-python-lib/_cm.json b/cm-mlops/script/get-generic-python-lib/_cm.json
@@ -161,7 +161,8 @@
           "names": [
             "cuda"
           ],
-          "tags": "get,cuda"
+          "tags": "get,cuda",
+          "version": "12.1.1"
         },
         {
           "names": [