Advance docker release latest verison to cuda 12.4 (#145566)

atalman · pytorchmergebot · commit 5d24a9a274d7 · 2025-01-24T15:27:25.000Z
Fixed latest tag in ghcr.io to be cuda 12.4 docker image. Todo, Need to add it to : https://github.com/pytorch/builder/blob/main/CUDA_UPGRADE_GUIDE.MD Will need to check if we can automate this by introducing cuda_stable variable or something like this. Pull Request resolved: #145566 Approved by: https://github.com/nWEIdia, https://github.com/kit1980, https://github.com/malfet
diff --git a/.github/workflows/docker-release.yml b/.github/workflows/docker-release.yml
@@ -153,7 +153,7 @@ jobs:
           docker push ghcr.io/pytorch/pytorch-nightly:"${PYTORCH_NIGHTLY_COMMIT}${CUDA_SUFFIX}"
 
           # Please note, here we ned to pin specific verison of CUDA as with latest label
-          if [[ ${CUDA_VERSION_SHORT} == "12.1" ]]; then
+          if [[ ${CUDA_VERSION_SHORT} == "12.4" ]]; then
             docker tag ghcr.io/pytorch/pytorch-nightly:"${PYTORCH_NIGHTLY_COMMIT}${CUDA_SUFFIX}" \
                     ghcr.io/pytorch/pytorch-nightly:latest
             docker push ghcr.io/pytorch/pytorch-nightly:latest
