Skip to content

Commit 59db0cd

Browse files
authored
Add a dev image for cuda12.3 (#6835)
1 parent d99ceb6 commit 59db0cd

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

infra/ansible/config/cuda_deps.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,14 @@
33
cuda_deps:
44
# List all libcudnn8 versions with `apt list -a libcudnn8`
55
libcudnn:
6+
"12.3": libcudnn9-cuda-12=9.0.0.312-1
67
"12.1": libcudnn8=8.9.2.26-1+cuda12.1
78
"12.0": libcudnn8=8.8.0.121-1+cuda12.0
89
"11.8": libcudnn8=8.7.0.84-1+cuda11.8
910
"11.7": libcudnn8=8.5.0.96-1+cuda11.7
1011
"11.2": libcudnn8=8.1.1.33-1+cuda11.2
1112
libcudnn-dev:
13+
"12.3": libcudnn9-dev-cuda-12=9.0.0.312-1
1214
"12.1": libcudnn8-dev=8.9.2.26-1+cuda12.1
1315
"12.0": libcudnn8-dev=8.8.0.121-1+cuda12.0
1416
"11.8": libcudnn8-dev=8.7.0.84-1+cuda11.8

infra/tpu-pytorch-releases/dev_images.auto.tfvars

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,5 +13,11 @@ dev_images = [
1313
accelerator = "cuda"
1414
cuda_version = "12.1"
1515
extra_tags = ["cuda"]
16+
},
17+
{
18+
accelerator = "cuda"
19+
cuda_version = "12.3"
20+
extra_tags = ["cuda"]
21+
python_version = "3.10"
1622
}
1723
]

0 commit comments

Comments
 (0)