@@ -43,11 +43,11 @@ jobs:
4343 include :
4444 # We only release one docker image per PyTorch version.
4545 # Make sure the matrix here matches the one below.
46- - { python_version: "3.10", pytorch_version: "2.1", cuda_version: "12.1.0 " }
47- - { python_version: "3.11", pytorch_version: "2.2", cuda_version: "12.1.0 " }
48- - { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.0 " }
49- - { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.0 " }
50- - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0 ", latest: "true" }
46+ - { python_version: "3.10", pytorch_version: "2.1", cuda_version: "12.1.1 " }
47+ - { python_version: "3.11", pytorch_version: "2.2", cuda_version: "12.1.1 " }
48+ - { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.1 " }
49+ - { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.1 " }
50+ - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.1 ", latest: "true" }
5151 steps :
5252 - uses : actions/checkout@v4
5353 with :
@@ -95,11 +95,11 @@ jobs:
9595 include :
9696 # These are the base images for PL release docker images.
9797 # Make sure the matrix here matches the one above.
98- - { python_version: "3.10", pytorch_version: "2.1.2", cuda_version: "12.1.0 " }
99- - { python_version: "3.11", pytorch_version: "2.2.2", cuda_version: "12.1.0 " }
100- - { python_version: "3.11", pytorch_version: "2.3.1", cuda_version: "12.1.0 " }
101- - { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.0 " }
102- - { python_version: "3.12", pytorch_version: "2.5.1", cuda_version: "12.1.0 " }
98+ - { python_version: "3.10", pytorch_version: "2.1.2", cuda_version: "12.1.1 " }
99+ - { python_version: "3.11", pytorch_version: "2.2.2", cuda_version: "12.1.1 " }
100+ - { python_version: "3.11", pytorch_version: "2.3.1", cuda_version: "12.1.1 " }
101+ - { python_version: "3.11", pytorch_version: "2.4.1", cuda_version: "12.1.1 " }
102+ - { python_version: "3.12", pytorch_version: "2.5.1", cuda_version: "12.1.1 " }
103103 steps :
104104 - uses : actions/checkout@v4
105105 - uses : docker/setup-buildx-action@v3
0 commit comments