Skip to content

Commit 37a514d

Browse files
author
Vincent Moens
committed
[CI] Cuda 12.4
ghstack-source-id: 2f3842a Pull Request resolved: #2733
1 parent fe3f00c commit 37a514d

File tree

6 files changed

+33
-33
lines changed

6 files changed

+33
-33
lines changed

‎.github/workflows/docs.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ jobs:
2727
strategy:
2828
matrix:
2929
python_version: ["3.10"]
30-
cuda_arch_version: ["12.1"]
30+
cuda_arch_version: ["12.4"]
3131
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
3232
with:
3333
repository: pytorch/rl

‎.github/workflows/test-linux-habitat.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ jobs:
2424
strategy:
2525
matrix:
2626
python_version: ["3.9"]
27-
cuda_arch_version: ["12.1"]
27+
cuda_arch_version: ["12.4"]
2828
fail-fast: false
2929
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
3030
with:

‎.github/workflows/test-linux-libs.yml

Lines changed: 27 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ jobs:
2525
strategy:
2626
matrix:
2727
python_version: ["3.9"]
28-
cuda_arch_version: ["12.1"]
28+
cuda_arch_version: ["12.4"]
2929
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
3030
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
3131
with:
@@ -59,7 +59,7 @@ jobs:
5959
strategy:
6060
matrix:
6161
python_version: ["3.11"]
62-
cuda_arch_version: ["12.1"]
62+
cuda_arch_version: ["12.4"]
6363
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
6464
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
6565
with:
@@ -96,7 +96,7 @@ jobs:
9696
strategy:
9797
matrix:
9898
python_version: ["3.9"]
99-
cuda_arch_version: ["12.1"]
99+
cuda_arch_version: ["12.4"]
100100
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
101101
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
102102
with:
@@ -131,7 +131,7 @@ jobs:
131131
strategy:
132132
matrix:
133133
python_version: ["3.9"]
134-
cuda_arch_version: ["12.1"]
134+
cuda_arch_version: ["12.4"]
135135
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
136136
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
137137
with:
@@ -166,7 +166,7 @@ jobs:
166166
strategy:
167167
matrix:
168168
python_version: ["3.9"]
169-
cuda_arch_version: ["12.1"]
169+
cuda_arch_version: ["12.4"]
170170
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
171171
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
172172
with:
@@ -200,7 +200,7 @@ jobs:
200200
strategy:
201201
matrix:
202202
python_version: ["3.9"]
203-
cuda_arch_version: ["12.1"]
203+
cuda_arch_version: ["12.4"]
204204
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
205205
with:
206206
repository: pytorch/rl
@@ -235,7 +235,7 @@ jobs:
235235
strategy:
236236
matrix:
237237
python_version: ["3.9"]
238-
cuda_arch_version: ["12.1"]
238+
cuda_arch_version: ["12.4"]
239239
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
240240
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
241241
with:
@@ -256,7 +256,7 @@ jobs:
256256
257257
set -euo pipefail
258258
export PYTHON_VERSION="3.9"
259-
export CU_VERSION="12.1"
259+
export CU_VERSION="12.4"
260260
export TAR_OPTIONS="--no-same-owner"
261261
export UPLOAD_CHANNEL="nightly"
262262
export TF_CPP_MIN_LOG_LEVEL=0
@@ -277,7 +277,7 @@ jobs:
277277
repository: pytorch/rl
278278
runner: "linux.g5.4xlarge.nvidia.gpu"
279279
gpu-arch-type: cuda
280-
gpu-arch-version: "12.1"
280+
gpu-arch-version: "12.4"
281281
docker-image: "nvidia/cuda:12.4.1-runtime-ubuntu22.04"
282282
timeout: 120
283283
script: |
@@ -291,7 +291,7 @@ jobs:
291291
292292
set -euo pipefail
293293
export PYTHON_VERSION="3.11"
294-
export CU_VERSION="12.1"
294+
export CU_VERSION="12.4"
295295
export TAR_OPTIONS="--no-same-owner"
296296
export UPLOAD_CHANNEL="nightly"
297297
export TF_CPP_MIN_LOG_LEVEL=0
@@ -309,7 +309,7 @@ jobs:
309309
strategy:
310310
matrix:
311311
python_version: ["3.9"]
312-
cuda_arch_version: ["12.1"]
312+
cuda_arch_version: ["12.4"]
313313
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
314314
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
315315
with:
@@ -330,7 +330,7 @@ jobs:
330330
331331
set -euo pipefail
332332
export PYTHON_VERSION="3.9"
333-
export CU_VERSION="12.1"
333+
export CU_VERSION="12.4"
334334
export TAR_OPTIONS="--no-same-owner"
335335
export UPLOAD_CHANNEL="nightly"
336336
export TF_CPP_MIN_LOG_LEVEL=0
@@ -347,7 +347,7 @@ jobs:
347347
strategy:
348348
matrix:
349349
python_version: ["3.9"]
350-
cuda_arch_version: ["12.1"]
350+
cuda_arch_version: ["12.4"]
351351
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
352352
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
353353
with:
@@ -368,7 +368,7 @@ jobs:
368368
369369
set -euo pipefail
370370
export PYTHON_VERSION="3.9"
371-
export CU_VERSION="12.1"
371+
export CU_VERSION="12.4"
372372
export TAR_OPTIONS="--no-same-owner"
373373
export UPLOAD_CHANNEL="nightly"
374374
export TF_CPP_MIN_LOG_LEVEL=0
@@ -385,7 +385,7 @@ jobs:
385385
strategy:
386386
matrix:
387387
python_version: ["3.10.12"]
388-
cuda_arch_version: ["12.1"]
388+
cuda_arch_version: ["12.4"]
389389
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
390390
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
391391
with:
@@ -406,7 +406,7 @@ jobs:
406406
407407
set -euo pipefail
408408
export PYTHON_VERSION="3.10.12"
409-
export CU_VERSION="12.1"
409+
export CU_VERSION="12.4"
410410
export TAR_OPTIONS="--no-same-owner"
411411
export UPLOAD_CHANNEL="nightly"
412412
export TF_CPP_MIN_LOG_LEVEL=0
@@ -423,7 +423,7 @@ jobs:
423423
strategy:
424424
matrix:
425425
python_version: ["3.9"]
426-
cuda_arch_version: ["12.1"]
426+
cuda_arch_version: ["12.4"]
427427
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
428428
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
429429
with:
@@ -458,7 +458,7 @@ jobs:
458458
strategy:
459459
matrix:
460460
python_version: ["3.9"]
461-
cuda_arch_version: ["12.1"]
461+
cuda_arch_version: ["12.4"]
462462
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
463463
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
464464
with:
@@ -510,7 +510,7 @@ jobs:
510510
511511
set -euo pipefail
512512
export PYTHON_VERSION="3.9"
513-
export CU_VERSION="12.1"
513+
export CU_VERSION="12.4"
514514
export TAR_OPTIONS="--no-same-owner"
515515
export UPLOAD_CHANNEL="nightly"
516516
export TF_CPP_MIN_LOG_LEVEL=0
@@ -528,7 +528,7 @@ jobs:
528528
strategy:
529529
matrix:
530530
python_version: ["3.9"]
531-
cuda_arch_version: ["12.1"]
531+
cuda_arch_version: ["12.4"]
532532
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
533533
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
534534
with:
@@ -562,7 +562,7 @@ jobs:
562562
strategy:
563563
matrix:
564564
python_version: ["3.9"]
565-
cuda_arch_version: ["12.1"]
565+
cuda_arch_version: ["12.4"]
566566
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
567567
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
568568
with:
@@ -597,7 +597,7 @@ jobs:
597597
strategy:
598598
matrix:
599599
python_version: ["3.9"]
600-
cuda_arch_version: ["12.1"]
600+
cuda_arch_version: ["12.4"]
601601
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
602602
with:
603603
repository: pytorch/rl
@@ -633,7 +633,7 @@ jobs:
633633
strategy:
634634
matrix:
635635
python_version: ["3.9"]
636-
cuda_arch_version: ["12.1"]
636+
cuda_arch_version: ["12.4"]
637637
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
638638
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
639639
with:
@@ -654,7 +654,7 @@ jobs:
654654
655655
set -euo pipefail
656656
export PYTHON_VERSION="3.9"
657-
export CU_VERSION="12.1"
657+
export CU_VERSION="12.4"
658658
export TAR_OPTIONS="--no-same-owner"
659659
export UPLOAD_CHANNEL="nightly"
660660
export TF_CPP_MIN_LOG_LEVEL=0
@@ -672,7 +672,7 @@ jobs:
672672
strategy:
673673
matrix:
674674
python_version: ["3.9"]
675-
cuda_arch_version: ["12.1"]
675+
cuda_arch_version: ["12.4"]
676676
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
677677
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
678678
with:
@@ -707,7 +707,7 @@ jobs:
707707
strategy:
708708
matrix:
709709
python_version: ["3.9"]
710-
cuda_arch_version: ["12.1"]
710+
cuda_arch_version: ["12.4"]
711711
if: ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
712712
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
713713
with:
@@ -728,7 +728,7 @@ jobs:
728728
729729
set -euo pipefail
730730
export PYTHON_VERSION="3.9"
731-
export CU_VERSION="12.1"
731+
export CU_VERSION="12.4"
732732
export TAR_OPTIONS="--no-same-owner"
733733
export UPLOAD_CHANNEL="nightly"
734734
export TF_CPP_MIN_LOG_LEVEL=0

‎.github/workflows/test-linux-rlhf.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ jobs:
2424
strategy:
2525
matrix:
2626
python_version: ["3.9"]
27-
cuda_arch_version: ["12.1"]
27+
cuda_arch_version: ["12.4"]
2828
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
2929
with:
3030
repository: pytorch/rl

‎.github/workflows/test-linux-sota.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ jobs:
2727
strategy:
2828
matrix:
2929
python_version: ["3.9"]
30-
cuda_arch_version: ["12.1"]
30+
cuda_arch_version: ["12.4"]
3131
fail-fast: false
3232
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
3333
with:

‎.github/workflows/test-linux.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ jobs:
8989
strategy:
9090
matrix:
9191
python_version: ["3.11"]
92-
cuda_arch_version: ["12.1"]
92+
cuda_arch_version: ["12.4"]
9393
fail-fast: false
9494
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
9595
with:
@@ -158,7 +158,7 @@ jobs:
158158
strategy:
159159
matrix:
160160
python_version: ["3.11"]
161-
cuda_arch_version: ["12.1"]
161+
cuda_arch_version: ["12.4"]
162162
fail-fast: false
163163
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
164164
with:

0 commit comments

Comments
 (0)