Skip to content

Commit 6a1ccfd

Browse files
committed
Add support for Linux cu124 build
1 parent b528bf3 commit 6a1ccfd

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

.github/workflows/cuda/Linux.sh

+6
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,12 @@ CUDA_VER_ID="${CUDA_VER_ARR[0]}_${CUDA_VER_ARR[1]}"
1212
CUDA_VER_SHORT="cu${CUDA_VER_ARR[0]}${CUDA_VER_ARR[1]}"
1313

1414
case ${CUDA_VER_SHORT} in
15+
cu124)
16+
CUDA=12.4
17+
APT_KEY=${OS}-${CUDA/./-}-local
18+
FILENAME=cuda-repo-${APT_KEY}_${CUDA}.1-550.54.15-1_amd64.deb
19+
URL=https://developer.download.nvidia.com/compute/cuda/${CUDA}.1/local_installers
20+
;;
1521
cu121)
1622
CUDA=12.1
1723
APT_KEY=${OS}-${CUDA/./-}-local

.github/workflows/ubuntu.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ jobs:
1919
os: [ubuntu-22.04, ubuntu-20.04] # [ubuntu-22.04, ubuntu-20.04, ubuntu-18.04]
2020
arch: [x64] # [x64, x86]
2121
torch-version: [2.6.0, 2.3.1, 2.2.1, 2.1.2] # [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.1.1, 2.1.2, 2.2.0, 2.2.1]
22-
cuda-version: [11.8.0, 12.4.99] # [12.3.1, 12.1.1, 11.8.0, 11.7.1, 11.6.2, 11.5.2,11.4.4, 11.3.1, 11.2.2, 11.1.1, 11.0.3, cpu]
22+
cuda-version: [11.8.0, 12.4.1] # [12.3.1, 12.1.1, 11.8.0, 11.7.1, 11.6.2, 11.5.2,11.4.4, 11.3.1, 11.2.2, 11.1.1, 11.0.3, cpu]
2323
cmake-build-type: [Release] # [Debug, ClangTidy]
2424
env:
2525
CCACHE_DIR: ${{ github.workspace }}/ccache

0 commit comments

Comments
 (0)