Skip to content

Commit 6068856

Browse files
vmoensvmoens
authored
[CI] Fix cuda versions (#889)
1 parent c6f60e4 commit 6068856

File tree

1 file changed

+33
-33
lines changed

1 file changed

+33
-33
lines changed

‎.circleci/config.yml

Lines changed: 33 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ binary_common: &binary_common
110110
wheel_docker_image:
111111
description: "Wheel only: what docker image to use"
112112
type: string
113-
default: "pytorch/manylinux-cuda113"
113+
default: "pytorch/manylinux-cuda116"
114114
conda_docker_image:
115115
description: "Conda only: what docker image to use"
116116
type: string
@@ -262,7 +262,7 @@ jobs:
262262
<<: *binary_common
263263

264264
docker:
265-
- image: "pytorch/manylinux-cuda113"
265+
- image: "pytorch/manylinux-cuda117"
266266
resource_class: 2xlarge+
267267

268268
environment:
@@ -315,7 +315,7 @@ jobs:
315315
image: ubuntu-2004-cuda-11.4:202110-01
316316
resource_class: gpu.nvidia.medium
317317
environment:
318-
image_name: "pytorch/manylinux-cuda113"
318+
image_name: "pytorch/manylinux-cuda117"
319319
TAR_OPTIONS: --no-same-owner
320320
PYTHON_VERSION: << parameters.python_version >>
321321
CU_VERSION: << parameters.cu_version >>
@@ -375,7 +375,7 @@ jobs:
375375
image: ubuntu-2004-cuda-11.4:202110-01
376376
resource_class: gpu.nvidia.medium
377377
environment:
378-
image_name: "pytorch/manylinux-cuda113"
378+
image_name: "pytorch/manylinux-cuda117"
379379
TAR_OPTIONS: --no-same-owner
380380
PYTHON_VERSION: << parameters.python_version >>
381381
CU_VERSION: << parameters.cu_version >>
@@ -480,7 +480,7 @@ jobs:
480480
image: ubuntu-2004-cuda-11.4:202110-01
481481
resource_class: gpu.nvidia.medium
482482
environment:
483-
image_name: "pytorch/manylinux-cuda113"
483+
image_name: "pytorch/manylinux-cuda117"
484484
TAR_OPTIONS: --no-same-owner
485485
PYTHON_VERSION: << parameters.python_version >>
486486
CU_VERSION: << parameters.cu_version >>
@@ -525,7 +525,7 @@ jobs:
525525
image: ubuntu-2004-cuda-11.4:202110-01
526526
resource_class: gpu.nvidia.medium
527527
environment:
528-
image_name: "pytorch/manylinux-cuda113"
528+
image_name: "pytorch/manylinux-cuda117"
529529
TAR_OPTIONS: --no-same-owner
530530
PYTHON_VERSION: << parameters.python_version >>
531531
CU_VERSION: << parameters.cu_version >>
@@ -570,7 +570,7 @@ jobs:
570570
image: ubuntu-2004-cuda-11.4:202110-01
571571
resource_class: gpu.nvidia.medium
572572
environment:
573-
image_name: "pytorch/manylinux-cuda113"
573+
image_name: "pytorch/manylinux-cuda117"
574574
TAR_OPTIONS: --no-same-owner
575575
PYTHON_VERSION: << parameters.python_version >>
576576
CU_VERSION: << parameters.cu_version >>
@@ -615,7 +615,7 @@ jobs:
615615
image: ubuntu-2004-cuda-11.4:202110-01
616616
resource_class: gpu.nvidia.medium
617617
environment:
618-
image_name: "pytorch/manylinux-cuda113"
618+
image_name: "pytorch/manylinux-cuda117"
619619
TAR_OPTIONS: --no-same-owner
620620
PYTHON_VERSION: << parameters.python_version >>
621621
CU_VERSION: << parameters.cu_version >>
@@ -660,7 +660,7 @@ jobs:
660660
image: ubuntu-2004-cuda-11.4:202110-01
661661
resource_class: gpu.nvidia.medium
662662
environment:
663-
image_name: "pytorch/manylinux-cuda113"
663+
image_name: "pytorch/manylinux-cuda117"
664664
TAR_OPTIONS: --no-same-owner
665665
PYTHON_VERSION: << parameters.python_version >>
666666
CU_VERSION: << parameters.cu_version >>
@@ -703,7 +703,7 @@ jobs:
703703
image: ubuntu-2004-cuda-11.4:202110-01
704704
resource_class: gpu.nvidia.medium
705705
environment:
706-
image_name: "pytorch/manylinux-cuda113"
706+
image_name: "pytorch/manylinux-cuda117"
707707
TAR_OPTIONS: --no-same-owner
708708
PYTHON_VERSION: << parameters.python_version >>
709709
CU_VERSION: << parameters.cu_version >>
@@ -760,7 +760,7 @@ jobs:
760760
<<: *binary_common
761761

762762
docker:
763-
- image: "pytorch/manylinux-cuda113"
763+
- image: "pytorch/manylinux-cuda116"
764764
resource_class: 2xlarge+
765765

766766
environment:
@@ -813,7 +813,7 @@ jobs:
813813
image: ubuntu-2004-cuda-11.4:202110-01
814814
resource_class: gpu.nvidia.medium
815815
environment:
816-
image_name: "pytorch/manylinux-cuda113"
816+
image_name: "pytorch/manylinux-cuda116"
817817
TAR_OPTIONS: --no-same-owner
818818
PYTHON_VERSION: << parameters.python_version >>
819819
CU_VERSION: << parameters.cu_version >>
@@ -872,7 +872,7 @@ jobs:
872872
image: ubuntu-2004-cuda-11.4:202110-01
873873
resource_class: gpu.nvidia.medium
874874
environment:
875-
image_name: "pytorch/manylinux-cuda113"
875+
image_name: "pytorch/manylinux-cuda116"
876876
TAR_OPTIONS: --no-same-owner
877877
PYTHON_VERSION: << parameters.python_version >>
878878
CU_VERSION: << parameters.cu_version >>
@@ -1178,19 +1178,19 @@ workflows:
11781178
name: unittest_linux_cpu_py3.7
11791179
python_version: '3.7'
11801180
- unittest_linux_gpu:
1181-
cu_version: cu113
1181+
cu_version: cu117
11821182
name: unittest_linux_gpu_py3.7
11831183
python_version: '3.7'
11841184
- unittest_linux_optdeps_gpu:
1185-
cu_version: cu113
1185+
cu_version: cu117
11861186
name: unittest_linux_optdeps_gpu_py3.7
11871187
python_version: '3.7'
11881188
- unittest_linux_stable_cpu:
11891189
cu_version: cpu
11901190
name: unittest_linux_stable_cpu_py3.7
11911191
python_version: '3.7'
11921192
- unittest_linux_stable_gpu:
1193-
cu_version: cu113
1193+
cu_version: cu116
11941194
name: unittest_linux_stable_gpu_py3.7
11951195
python_version: '3.7'
11961196

@@ -1203,44 +1203,44 @@ workflows:
12031203
name: unittest_linux_cpu_py3.8
12041204
python_version: '3.8'
12051205
- unittest_linux_gpu:
1206-
cu_version: cu113
1206+
cu_version: cu117
12071207
name: unittest_linux_gpu_py3.8
12081208
python_version: '3.8'
12091209
- unittest_linux_optdeps_gpu:
1210-
cu_version: cu113
1210+
cu_version: cu117
12111211
name: unittest_linux_optdeps_gpu_py3.8
12121212
python_version: '3.8'
12131213
- unittest_linux_stable_cpu:
12141214
cu_version: cpu
12151215
name: unittest_linux_stable_cpu_py3.8
12161216
python_version: '3.8'
12171217
- unittest_linux_stable_gpu:
1218-
cu_version: cu113
1218+
cu_version: cu116
12191219
name: unittest_linux_stable_gpu_py3.8
12201220
python_version: '3.8'
12211221
# we test supported libs for 3.8 only
12221222
- unittest_linux_habitat_gpu:
1223-
cu_version: cu113
1223+
cu_version: cu117
12241224
name: unittest_linux_habitat_gpu_py3.8
12251225
python_version: '3.8'
12261226
- unittest_linux_jumanji_gpu:
1227-
cu_version: cu113
1227+
cu_version: cu117
12281228
name: unittest_linux_jumanji_gpu_py3.8
12291229
python_version: '3.8'
12301230
- unittest_linux_envpool_gpu:
1231-
cu_version: cu113
1231+
cu_version: cu117
12321232
name: unittest_linux_envpool_gpu_py3.8
12331233
python_version: '3.8'
12341234
- unittest_linux_brax_gpu:
1235-
cu_version: cu113
1235+
cu_version: cu117
12361236
name: unittest_linux_brax_gpu_py3.8
12371237
python_version: '3.8'
12381238
- unittest_linux_vmas_gpu:
1239-
cu_version: cu113
1239+
cu_version: cu117
12401240
name: unittest_linux_vmas_gpu_py3.8
12411241
python_version: '3.8'
12421242
- unittest_linux_gym_gpu:
1243-
cu_version: cu113
1243+
cu_version: cu117
12441244
name: unittest_linux_gym_gpu_py3.8
12451245
python_version: '3.8'
12461246

@@ -1254,19 +1254,19 @@ workflows:
12541254
name: unittest_linux_cpu_py3.9
12551255
python_version: '3.9'
12561256
- unittest_linux_gpu:
1257-
cu_version: cu113
1257+
cu_version: cu117
12581258
name: unittest_linux_gpu_py3.9
12591259
python_version: '3.9'
12601260
- unittest_linux_optdeps_gpu:
1261-
cu_version: cu113
1261+
cu_version: cu117
12621262
name: unittest_linux_optdeps_gpu_py3.9
12631263
python_version: '3.9'
12641264
- unittest_linux_stable_cpu:
12651265
cu_version: cpu
12661266
name: unittest_linux_stable_cpu_py3.9
12671267
python_version: '3.9'
12681268
- unittest_linux_stable_gpu:
1269-
cu_version: cu113
1269+
cu_version: cu116
12701270
name: unittest_linux_stable_gpu_py3.9
12711271
python_version: '3.9'
12721272

@@ -1279,29 +1279,29 @@ workflows:
12791279
name: unittest_linux_cpu_py3.10
12801280
python_version: '3.10'
12811281
- unittest_linux_gpu:
1282-
cu_version: cu113
1282+
cu_version: cu117
12831283
name: unittest_linux_gpu_py3.10
12841284
python_version: '3.10'
12851285
- unittest_linux_optdeps_gpu:
1286-
cu_version: cu113
1286+
cu_version: cu117
12871287
name: unittest_linux_optdeps_gpu_py3.10
12881288
python_version: '3.10'
12891289
- unittest_linux_stable_cpu:
12901290
cu_version: cpu
12911291
name: unittest_linux_stable_cpu_py3.10
12921292
python_version: '3.10'
12931293
- unittest_linux_stable_gpu:
1294-
cu_version: cu113
1294+
cu_version: cu116
12951295
name: unittest_linux_stable_gpu_py3.10
12961296
python_version: '3.10'
12971297

12981298
- unittest_linux_olddeps_gpu:
1299-
cu_version: cu113
1299+
cu_version: cu116
13001300
name: unittest_linux_olddeps_gpu_py3.9
13011301
python_version: '3.9'
13021302

13031303
- unittest_linux_examples_gpu:
1304-
cu_version: cu113
1304+
cu_version: cu117
13051305
name: unittest_linux_examples_gpu_py3.9
13061306
python_version: '3.9'
13071307

0 commit comments

Comments
 (0)