Skip to content

Commit b9aec01

Browse files
atalmanatalman
committed
Add cuda.z version to the start page
1 parent 05a0c25 commit b9aec01

File tree

3 files changed

+61
-9
lines changed

3 files changed

+61
-9
lines changed

‎_includes/quick_start_local.html

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,10 +88,13 @@
8888
<div class="option-text">Compute Platform</div>
8989
</div>
9090
<div class="col-md-3 option block version" id="cuda.x">
91-
<div class="option-text">CUDA 11.6</div>
91+
<div class="option-text">CUDA 11.8</div>
9292
</div>
9393
<div class="col-md-3 option block version" id="cuda.y">
94-
<div class="option-text">CUDA 11.7</div>
94+
<div class="option-text">CUDA 12.1</div>
95+
</div>
96+
<div class="col-md-3 option block version" id="cuda.z">
97+
<div class="option-text">CUDA 12.4</div>
9598
</div>
9699
<div class="col-md-3 option block version" id="rocm5.x">
97100
<div class="option-text">ROCm 5.2</div>

‎published_versions.json

Lines changed: 48 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,10 @@
1717
"note": null,
1818
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121"
1919
},
20+
"cuda.z": {
21+
"note": null,
22+
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124"
23+
},
2024
"rocm5.x": {
2125
"note": null,
2226
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0"
@@ -31,6 +35,10 @@
3135
"note": null,
3236
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia"
3337
},
38+
"cuda.z": {
39+
"note": null,
40+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia"
41+
},
3442
"rocm5.x": {
3543
"note": "<b>NOTE:</b> Conda packages are not currently available for ROCm, please use pip instead<br />",
3644
"command": null
@@ -62,6 +70,13 @@
6270
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip"
6371
}
6472
},
73+
"cuda.z": {
74+
"note": null,
75+
"versions": {
76+
"Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-shared-with-deps-latest.zip",
77+
"Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-cxx11-abi-shared-with-deps-latest.zip"
78+
}
79+
},
6580
"rocm5.x": {
6681
"note": null,
6782
"versions": {
@@ -83,6 +98,11 @@
8398
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
8499
"default": true
85100
},
101+
"cuda.z": {
102+
"note": "# CUDA is not available on MacOS, please use default package",
103+
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
104+
"default": true
105+
},
86106
"rocm5.x": {
87107
"note": "# ROCm is not available on MacOS, please use default package",
88108
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
@@ -104,6 +124,11 @@
104124
"command": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly",
105125
"default": true
106126
},
127+
"cuda.z": {
128+
"note": "# CUDA is not available on MacOS, please use default package",
129+
"command": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly",
130+
"default": true
131+
},
107132
"rocm5.x": {
108133
"note": "# ROCm is not available on MacOS, please use default package",
109134
"command": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly",
@@ -135,6 +160,13 @@
135160
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip"
136161
}
137162
},
163+
"cuda.z": {
164+
"note": null,
165+
"default": true,
166+
"versions": {
167+
"Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip"
168+
}
169+
},
138170
"rocm5.x": {
139171
"note": null,
140172
"default": true,
@@ -158,6 +190,10 @@
158190
"note": null,
159191
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121"
160192
},
193+
"cuda.z": {
194+
"note": null,
195+
"command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124"
196+
},
161197
"rocm5.x": {
162198
"note": "<b>NOTE:</b> ROCm is not available on Windows",
163199
"command": null
@@ -172,6 +208,10 @@
172208
"note": null,
173209
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia"
174210
},
211+
"cuda.z": {
212+
"note": null,
213+
"command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia"
214+
},
175215
"rocm5.x": {
176216
"note": "<b>NOTE:</b> ROCm is not available on Windows",
177217
"command": null
@@ -203,6 +243,13 @@
203243
"Download here (Debug version):": "https://download.pytorch.org/libtorch/nightly/cu121/libtorch-win-shared-with-deps-debug-latest.zip"
204244
}
205245
},
246+
"cuda.z": {
247+
"note": null,
248+
"versions": {
249+
"Download here (Release version):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-win-shared-with-deps-latest.zip",
250+
"Download here (Debug version):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-win-shared-with-deps-debug-latest.zip"
251+
}
252+
},
206253
"rocm5.x": {
207254
"note": "<b>NOTE:</b> ROCm is not available on Windows",
208255
"versions": null
@@ -4335,4 +4382,4 @@
43354382
}
43364383
}
43374384
}
4338-
}
4385+
}

‎scripts/gen_quick_start_module.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,14 @@ class OperatingSystem(Enum):
3737
"accnone": ("cpu", ""),
3838
"cuda.x": ("cuda", "11.8"),
3939
"cuda.y": ("cuda", "12.1"),
40+
"cuda.z": ("cuda", "12.4"),
4041
"rocm5.x": ("rocm", "6.0")
4142
},
4243
"release": {
4344
"accnone": ("cpu", ""),
4445
"cuda.x": ("cuda", "11.8"),
4546
"cuda.y": ("cuda", "12.1"),
47+
"cuda.z": ("cuda", "12.4"),
4648
"rocm5.x": ("rocm", "6.0")
4749
}
4850
}
@@ -184,16 +186,16 @@ def gen_ver_list(chan, gpu_arch_type):
184186
for chan in ("nightly", "release"):
185187
cuda_ver_list = gen_ver_list(chan, "cuda")
186188
rocm_ver_list = gen_ver_list(chan, "rocm")
187-
cuda_list = sorted(cuda_ver_list.values())[-2:]
189+
cuda_list = cuda_ver_list.values()
188190
acc_arch_ver_map[chan]["rocm5.x"] = ("rocm", max(rocm_ver_list.values()))
189-
for cuda_ver, label in zip(cuda_list, ["cuda.x", "cuda.y"]):
191+
for cuda_ver, label in zip(cuda_list, ["cuda.x", "cuda.y", "cuda.z"]):
190192
acc_arch_ver_map[chan][label] = ("cuda", cuda_ver)
191193

192194

193195
def main():
194196
parser = argparse.ArgumentParser()
195197
parser.add_argument('--autogenerate', dest='autogenerate', action='store_true')
196-
parser.set_defaults(autogenerate=False)
198+
parser.set_defaults(autogenerate=True)
197199

198200
options = parser.parse_args()
199201
versions = read_published_versions()
@@ -213,9 +215,9 @@ def main():
213215

214216
template = read_quick_start_module_template()
215217
versions_str = json.dumps(gen_install_matrix(versions))
216-
template = template.replace("{{ installMatrix }}", versions_str)
217-
template = template.replace("{{ VERSION }}", f"\"Stable ({versions['latest_stable']})\"")
218-
print(template.replace("{{ ACC ARCH MAP }}", json.dumps(acc_arch_ver_map)))
218+
#template = template.replace("{{ installMatrix }}", versions_str)
219+
#template = template.replace("{{ VERSION }}", f"\"Stable ({versions['latest_stable']})\"")
220+
#print(template.replace("{{ ACC ARCH MAP }}", json.dumps(acc_arch_ver_map)))
219221

220222
if __name__ == "__main__":
221223
main()

0 commit comments

Comments
 (0)