Skip to content

Commit ee9b9d1

Browse files
committed
use magma-cuda build-1 by default; remove CUDA 10.2-11.5 builds
1 parent 919cfc8 commit ee9b9d1

File tree

2 files changed

+5
-15
lines changed

2 files changed

+5
-15
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
runs-on: ubuntu-18.04
2929
strategy:
3030
matrix:
31-
cuda_version: ["11.8", "11.7", "11.6", "11.5", "11.3", "10.2"]
31+
cuda_version: ["11.8", "11.7", "11.6"]
3232
env:
3333
GPU_ARCH_TYPE: cuda
3434
GPU_ARCH_VERSION: ${{ matrix.cuda_version }}

common/install_magma.sh

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,10 @@ MAGMA_VERSION="2.5.2"
77
function do_install() {
88
cuda_version=$1
99
cuda_version_nodot=${1/./}
10-
11-
if [[ ${cuda_version_nodot} == 116 ]]; then
12-
MAGMA_VERSION="2.6.1"
13-
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-0.tar.bz2"
14-
elif [[ ${cuda_version_nodot} == 117 ]]; then
15-
MAGMA_VERSION="2.6.1"
16-
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-0.tar.bz2"
17-
elif [[ ${cuda_version_nodot} == 118 ]]; then
18-
MAGMA_VERSION="2.6.1"
19-
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
20-
else
21-
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
22-
fi
23-
10+
11+
MAGMA_VERSION="2.6.1"
12+
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"
13+
2414
cuda_dir="/usr/local/cuda-${cuda_version}"
2515
(
2616
set -x

0 commit comments

Comments
 (0)