25
25
strategy :
26
26
matrix :
27
27
python_version : ["3.9"]
28
- cuda_arch_version : ["12.1 "]
28
+ cuda_arch_version : ["12.4 "]
29
29
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
30
30
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
31
31
with :
59
59
strategy :
60
60
matrix :
61
61
python_version : ["3.11"]
62
- cuda_arch_version : ["12.1 "]
62
+ cuda_arch_version : ["12.4 "]
63
63
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
64
64
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
65
65
with :
96
96
strategy :
97
97
matrix :
98
98
python_version : ["3.9"]
99
- cuda_arch_version : ["12.1 "]
99
+ cuda_arch_version : ["12.4 "]
100
100
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
101
101
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
102
102
with :
@@ -131,7 +131,7 @@ jobs:
131
131
strategy :
132
132
matrix :
133
133
python_version : ["3.9"]
134
- cuda_arch_version : ["12.1 "]
134
+ cuda_arch_version : ["12.4 "]
135
135
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
136
136
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
137
137
with :
@@ -166,7 +166,7 @@ jobs:
166
166
strategy :
167
167
matrix :
168
168
python_version : ["3.9"]
169
- cuda_arch_version : ["12.1 "]
169
+ cuda_arch_version : ["12.4 "]
170
170
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
171
171
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
172
172
with :
@@ -200,7 +200,7 @@ jobs:
200
200
strategy :
201
201
matrix :
202
202
python_version : ["3.9"]
203
- cuda_arch_version : ["12.1 "]
203
+ cuda_arch_version : ["12.4 "]
204
204
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
205
205
with :
206
206
repository : pytorch/rl
@@ -235,7 +235,7 @@ jobs:
235
235
strategy :
236
236
matrix :
237
237
python_version : ["3.9"]
238
- cuda_arch_version : ["12.1 "]
238
+ cuda_arch_version : ["12.4 "]
239
239
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
240
240
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
241
241
with :
@@ -256,7 +256,7 @@ jobs:
256
256
257
257
set -euo pipefail
258
258
export PYTHON_VERSION="3.9"
259
- export CU_VERSION="12.1 "
259
+ export CU_VERSION="12.4 "
260
260
export TAR_OPTIONS="--no-same-owner"
261
261
export UPLOAD_CHANNEL="nightly"
262
262
export TF_CPP_MIN_LOG_LEVEL=0
@@ -277,7 +277,7 @@ jobs:
277
277
repository : pytorch/rl
278
278
runner : " linux.g5.4xlarge.nvidia.gpu"
279
279
gpu-arch-type : cuda
280
- gpu-arch-version : " 12.1 "
280
+ gpu-arch-version : " 12.4 "
281
281
docker-image : " nvidia/cuda:12.4.1-runtime-ubuntu22.04"
282
282
timeout : 120
283
283
script : |
@@ -291,7 +291,7 @@ jobs:
291
291
292
292
set -euo pipefail
293
293
export PYTHON_VERSION="3.11"
294
- export CU_VERSION="12.1 "
294
+ export CU_VERSION="12.4 "
295
295
export TAR_OPTIONS="--no-same-owner"
296
296
export UPLOAD_CHANNEL="nightly"
297
297
export TF_CPP_MIN_LOG_LEVEL=0
@@ -309,7 +309,7 @@ jobs:
309
309
strategy :
310
310
matrix :
311
311
python_version : ["3.9"]
312
- cuda_arch_version : ["12.1 "]
312
+ cuda_arch_version : ["12.4 "]
313
313
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
314
314
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
315
315
with :
@@ -330,7 +330,7 @@ jobs:
330
330
331
331
set -euo pipefail
332
332
export PYTHON_VERSION="3.9"
333
- export CU_VERSION="12.1 "
333
+ export CU_VERSION="12.4 "
334
334
export TAR_OPTIONS="--no-same-owner"
335
335
export UPLOAD_CHANNEL="nightly"
336
336
export TF_CPP_MIN_LOG_LEVEL=0
@@ -347,7 +347,7 @@ jobs:
347
347
strategy :
348
348
matrix :
349
349
python_version : ["3.9"]
350
- cuda_arch_version : ["12.1 "]
350
+ cuda_arch_version : ["12.4 "]
351
351
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
352
352
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
353
353
with :
@@ -368,7 +368,7 @@ jobs:
368
368
369
369
set -euo pipefail
370
370
export PYTHON_VERSION="3.9"
371
- export CU_VERSION="12.1 "
371
+ export CU_VERSION="12.4 "
372
372
export TAR_OPTIONS="--no-same-owner"
373
373
export UPLOAD_CHANNEL="nightly"
374
374
export TF_CPP_MIN_LOG_LEVEL=0
@@ -385,7 +385,7 @@ jobs:
385
385
strategy :
386
386
matrix :
387
387
python_version : ["3.10.12"]
388
- cuda_arch_version : ["12.1 "]
388
+ cuda_arch_version : ["12.4 "]
389
389
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
390
390
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
391
391
with :
@@ -406,7 +406,7 @@ jobs:
406
406
407
407
set -euo pipefail
408
408
export PYTHON_VERSION="3.10.12"
409
- export CU_VERSION="12.1 "
409
+ export CU_VERSION="12.4 "
410
410
export TAR_OPTIONS="--no-same-owner"
411
411
export UPLOAD_CHANNEL="nightly"
412
412
export TF_CPP_MIN_LOG_LEVEL=0
@@ -423,7 +423,7 @@ jobs:
423
423
strategy :
424
424
matrix :
425
425
python_version : ["3.9"]
426
- cuda_arch_version : ["12.1 "]
426
+ cuda_arch_version : ["12.4 "]
427
427
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
428
428
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
429
429
with :
@@ -458,7 +458,7 @@ jobs:
458
458
strategy :
459
459
matrix :
460
460
python_version : ["3.9"]
461
- cuda_arch_version : ["12.1 "]
461
+ cuda_arch_version : ["12.4 "]
462
462
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
463
463
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
464
464
with :
@@ -510,7 +510,7 @@ jobs:
510
510
511
511
set -euo pipefail
512
512
export PYTHON_VERSION="3.9"
513
- export CU_VERSION="12.1 "
513
+ export CU_VERSION="12.4 "
514
514
export TAR_OPTIONS="--no-same-owner"
515
515
export UPLOAD_CHANNEL="nightly"
516
516
export TF_CPP_MIN_LOG_LEVEL=0
@@ -528,7 +528,7 @@ jobs:
528
528
strategy :
529
529
matrix :
530
530
python_version : ["3.9"]
531
- cuda_arch_version : ["12.1 "]
531
+ cuda_arch_version : ["12.4 "]
532
532
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
533
533
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
534
534
with :
@@ -562,7 +562,7 @@ jobs:
562
562
strategy :
563
563
matrix :
564
564
python_version : ["3.9"]
565
- cuda_arch_version : ["12.1 "]
565
+ cuda_arch_version : ["12.4 "]
566
566
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
567
567
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
568
568
with :
@@ -597,7 +597,7 @@ jobs:
597
597
strategy :
598
598
matrix :
599
599
python_version : ["3.9"]
600
- cuda_arch_version : ["12.1 "]
600
+ cuda_arch_version : ["12.4 "]
601
601
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
602
602
with :
603
603
repository : pytorch/rl
@@ -633,7 +633,7 @@ jobs:
633
633
strategy :
634
634
matrix :
635
635
python_version : ["3.9"]
636
- cuda_arch_version : ["12.1 "]
636
+ cuda_arch_version : ["12.4 "]
637
637
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
638
638
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
639
639
with :
@@ -654,7 +654,7 @@ jobs:
654
654
655
655
set -euo pipefail
656
656
export PYTHON_VERSION="3.9"
657
- export CU_VERSION="12.1 "
657
+ export CU_VERSION="12.4 "
658
658
export TAR_OPTIONS="--no-same-owner"
659
659
export UPLOAD_CHANNEL="nightly"
660
660
export TF_CPP_MIN_LOG_LEVEL=0
@@ -672,7 +672,7 @@ jobs:
672
672
strategy :
673
673
matrix :
674
674
python_version : ["3.9"]
675
- cuda_arch_version : ["12.1 "]
675
+ cuda_arch_version : ["12.4 "]
676
676
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Data') }}
677
677
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
678
678
with :
@@ -707,7 +707,7 @@ jobs:
707
707
strategy :
708
708
matrix :
709
709
python_version : ["3.9"]
710
- cuda_arch_version : ["12.1 "]
710
+ cuda_arch_version : ["12.4 "]
711
711
if : ${{ github.event_name == 'push' || contains(github.event.pull_request.labels.*.name, 'Environments') }}
712
712
uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
713
713
with :
@@ -728,7 +728,7 @@ jobs:
728
728
729
729
set -euo pipefail
730
730
export PYTHON_VERSION="3.9"
731
- export CU_VERSION="12.1 "
731
+ export CU_VERSION="12.4 "
732
732
export TAR_OPTIONS="--no-same-owner"
733
733
export UPLOAD_CHANNEL="nightly"
734
734
export TF_CPP_MIN_LOG_LEVEL=0
0 commit comments