build libtorch and manywheel for 12.4 (#1723)

pytorch · Mar 15, 2024 · 60c856f · 60c856f
1 parent ab5fc90
commit 60c856f
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/.github/workflows/build-libtorch-images.yml b/.github/workflows/build-libtorch-images.yml
@@ -33,7 +33,7 @@ jobs:
     runs-on: linux.12xlarge.ephemeral
     strategy:
       matrix:
-        cuda_version: ["12.1", "11.8"]
+        cuda_version: ["12.4", "12.1", "11.8"]
     env:
       GPU_ARCH_TYPE: cuda
       GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

diff --git a/.github/workflows/build-manywheel-images.yml b/.github/workflows/build-manywheel-images.yml
@@ -37,7 +37,7 @@ jobs:
     runs-on: linux.12xlarge.ephemeral
     strategy:
       matrix:
-        cuda_version: ["12.1", "11.8"]
+        cuda_version: ["12.4", "12.1", "11.8"]
     env:
       GPU_ARCH_TYPE: cuda
       GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile
@@ -55,6 +55,11 @@ RUN bash ./install_cuda.sh 12.1
 RUN bash ./install_magma.sh 12.1
 RUN ln -sf /usr/local/cuda-12.1 /usr/local/cuda
 
+FROM cuda as cuda12.4
+RUN bash ./install_cuda.sh 12.4
+RUN bash ./install_magma.sh 12.4
+RUN ln -sf /usr/local/cuda-12.4 /usr/local/cuda
+
 FROM cpu as rocm
 ARG PYTORCH_ROCM_ARCH
 ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH}

diff --git a/libtorch/build_all_docker.sh b/libtorch/build_all_docker.sh
@@ -4,7 +4,7 @@ set -eou pipefail
 
 TOPDIR=$(git rev-parse --show-toplevel)
 
-for cuda_version in 12.1 11.8; do
+for cuda_version in 12.4 12.1 11.8; do
     GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/libtorch/build_docker.sh"
 done