Remove ubuntu as base image

NVIDIA · May 14, 2024 · 13d93a0 · 13d93a0
1 parent 08ff39f
commit 13d93a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/_build_base.yaml b/.github/workflows/_build_base.yaml
@@ -90,8 +90,8 @@ jobs:
               exit 1
             fi
             source .github/workflows/scripts/get_remote_env.sh
-            JAX_LIBNCCL_PACKAGE=$(get_remote_env ${BASE_IMAGE:-ubuntu} linux amd64 | jq -r '.[]' | egrep '^NV_LIBNCCL_PACKAGE')
-            JAX_NCCL_VERSION=$(get_remote_env ${BASE_IMAGE:-ubuntu} linux amd64 | jq -r '.[]' | egrep '^NCCL_VERSION=' | cut -d= -f2-)
+            JAX_LIBNCCL_PACKAGE=$(get_remote_env ${BASE_IMAGE} linux amd64 | jq -r '.[]' | egrep '^NV_LIBNCCL_PACKAGE')
+            JAX_NCCL_VERSION=$(get_remote_env ${BASE_IMAGE} linux amd64 | jq -r '.[]' | egrep '^NCCL_VERSION=' | cut -d= -f2-)
           else
             JAX_NCCL_VERSION=$(echo $JAX_LIBNCCL_PACKAGE | cut -d= -f2 | cut -d+ -f1)
           fi