fix hardcoded rapids ver

rapidsai · Mar 7, 2024 · 4b61a69 · 4b61a69
1 parent 4b93681
commit 4b61a69
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/release-to-ngc.yml b/.github/workflows/release-to-ngc.yml
@@ -50,10 +50,11 @@ jobs:
           CUDA_VER=${{ matrix.CUDA_VER }}
           CUDA_VER_SHORT=${CUDA_VER%.*}
           PYTHON_VER=${{ matrix.PYTHON_VER }}
+          RAPIDS_VER=24.02
 
           for type in base notebooks; do
-            source="rapidsai/$type:24.02-cuda$CUDA_VER_SHORT-py$PYTHON_VER"
-            target="nvcr.io/nvidia/rapidsai/$type:23.08-cuda$CUDA_VER_SHORT-py$PYTHON_VER"
+            source="rapidsai/$type:$RAPIDS_VER-cuda$CUDA_VER_SHORT-py$PYTHON_VER"
+            target="nvcr.io/nvidia/rapidsai/$type:$RAPIDS_VER-cuda$CUDA_VER_SHORT-py$PYTHON_VER"
             echo "$source => $target"
             docker run -v ~/.docker/config.json:/config.json quay.io/skopeo/stable:latest copy --multi-arch all --dest-authfile=/config.json docker://$source docker://$target
           done