add CUDA 11.7 builds for manywheel and libtorch (#1054)

* add CUDA 11.7 builds for manywheel and libtorch

* add libtorch and magma builds

Co-authored-by: pbialecki <[email protected]>

Author: ptrblck

.github/workflows/build-libtorch-images.yml

@@ -28,7 +28,7 @@ jobs:
     runs-on: ubuntu-18.04
     strategy:
       matrix:
-        cuda_version: ["11.6", "11.5", "11.3", "10.2"]
+        cuda_version: ["11.7", "11.6", "11.5", "11.3", "10.2"]
     env:
       GPU_ARCH_TYPE: cuda
       GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

.github/workflows/build-manywheel-images.yml

@@ -28,7 +28,7 @@ jobs:
     runs-on: ubuntu-18.04
     strategy:
       matrix:
-        cuda_version: ["11.6", "11.5", "11.3", "10.2"]
+        cuda_version: ["11.7", "11.6", "11.5", "11.3", "10.2"]
     env:
       GPU_ARCH_TYPE: cuda
       GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

common/install_magma.sh

@@ -11,6 +11,9 @@ function do_install() {
     if [[ ${cuda_version_nodot} == 116 ]]; then
         MAGMA_VERSION="2.6.1"
         magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-0.tar.bz2"
+    elif [[ ${cuda_version_nodot} == 117 ]]; then
+	MAGMA_VERSION="2.6.1"
+	magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-0.tar.bz2"
     else
         magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
     fi

libtorch/Dockerfile

@@ -60,6 +60,10 @@ FROM cuda as cuda11.6
 RUN bash ./install_cuda.sh 11.6
 RUN bash ./install_magma.sh 11.6
 
+FROM cuda as cuda11.7
+RUN bash ./install_cuda.sh 11.7
+RUN bash ./install_magma.sh 11.7
+
 FROM cpu as rocm
 ARG PYTORCH_ROCM_ARCH
 ENV PYTORCH_ROCM_ARCH ${PYTORCH_ROCM_ARCH}

libtorch/build_all_docker.sh

@@ -4,7 +4,7 @@ set -eou pipefail
 
 TOPDIR=$(git rev-parse --show-toplevel)
 
-for cuda_version in 11.6 11.5 11.3 10.2; do
+for cuda_version in 11.7 11.6 11.5 11.3 10.2; do
     GPU_ARCH_TYPE=cuda GPU_ARCH_VERSION="${cuda_version}" "${TOPDIR}/libtorch/build_docker.sh"
 done