Skip to content

Commit e57b2c1

Browse files
authored
Add nightly build for cuda 12 (#5253)
1 parent f6409de commit e57b2c1

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

infra/ansible/config/cuda_deps.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,12 @@
33
cuda_deps:
44
# List all libcudnn8 versions with `apt list -a libcudnn8`
55
libcudnn:
6+
"12.0": libcudnn8=8.8.0.121-1+cuda12.0
67
"11.8": libcudnn8=8.8.0.121-1+cuda11.8
78
"11.7": libcudnn8=8.5.0.96-1+cuda11.7
89
"11.2": libcudnn8=8.1.1.33-1+cuda11.2
910
libcudnn-dev:
11+
"12.0": libcudnn8-dev=8.8.0.121-1+cuda12.0
1012
"11.8": libcudnn8-dev=8.8.0.121-1+cuda11.8
1113
"11.7": libcudnn8-dev=8.5.0.96-1+cuda11.7
1214
"11.2": libcudnn8-dev=8.1.1.33-1+cuda11.2

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,10 @@ nightly_builds = [
77
accelerator = "tpu"
88
python_version = "3.10"
99
},
10+
{
11+
accelerator = "cuda"
12+
cuda_version = "12.0"
13+
},
1014
{
1115
accelerator = "cuda"
1216
cuda_version = "11.8"

0 commit comments

Comments
 (0)