diff --git a/.github/workflows/build-libtorch-images.yml b/.github/workflows/build-libtorch-images.yml index 742ca00c7..cf24caae6 100644 --- a/.github/workflows/build-libtorch-images.yml +++ b/.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 }} diff --git a/.github/workflows/build-manywheel-images.yml b/.github/workflows/build-manywheel-images.yml index ce5cc12d0..f2d6afb8d 100644 --- a/.github/workflows/build-manywheel-images.yml +++ b/.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 }} diff --git a/common/install_magma.sh b/common/install_magma.sh index ea2548140..5d14dbfe2 100644 --- a/common/install_magma.sh +++ b/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 diff --git a/libtorch/Dockerfile b/libtorch/Dockerfile index d5b22b09e..8ae734d2d 100644 --- a/libtorch/Dockerfile +++ b/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} diff --git a/libtorch/build_all_docker.sh b/libtorch/build_all_docker.sh index 1493778e4..208242805 100755 --- 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 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