Use pip install for nvidia-apex by default

ctuning · Feb 16, 2024 · 80dd969 · 80dd969
1 parent ea77418
commit 80dd969
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cm-mlops/script/get-generic-python-lib/_cm.json b/cm-mlops/script/get-generic-python-lib/_cm.json
@@ -149,6 +149,16 @@
       ]
     },
     "nvidia-apex": {
+      "env": {
+        "CM_GENERIC_PYTHON_PACKAGE_NAME": "apex",
+        "CM_GENERIC_PYTHON_PACKAGE_VARIANT": "nvidia-apex",
+        "CM_GENERIC_PYTHON_PIP_URL": "git+https://github.com/nvidia/apex@0da3ffb92ee6fbe5336602f0e3989db1cd16f880"
+      },
+      "new_env_keys": [
+        "CM_NVIDIA_APEX_VERSION"
+      ]
+    },
+    "nvidia-apex-from-src": {
       "env": {
         "CM_GENERIC_PYTHON_PACKAGE_NAME": "apex",
         "CM_GENERIC_PYTHON_PACKAGE_VARIANT": "nvidia-apex"