From b9aec01e56c514c2d7e28f51aff332049f6eb885 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 09:54:45 -0700 Subject: [PATCH 01/13] Add cuda.z version to the start page --- _includes/quick_start_local.html | 7 +++-- published_versions.json | 49 ++++++++++++++++++++++++++++++- scripts/gen_quick_start_module.py | 14 +++++---- 3 files changed, 61 insertions(+), 9 deletions(-) diff --git a/_includes/quick_start_local.html b/_includes/quick_start_local.html index 59ea52366dbb..06ad1f0e7ce5 100644 --- a/_includes/quick_start_local.html +++ b/_includes/quick_start_local.html @@ -88,10 +88,13 @@
Compute Platform
-
CUDA 11.6
+
CUDA 11.8
-
CUDA 11.7
+
CUDA 12.1
+
+
+
CUDA 12.4
ROCm 5.2
diff --git a/published_versions.json b/published_versions.json index ceab4c44ac0c..450967483a3c 100644 --- a/published_versions.json +++ b/published_versions.json @@ -17,6 +17,10 @@ "note": null, "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121" }, + "cuda.z": { + "note": null, + "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124" + }, "rocm5.x": { "note": null, "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0" @@ -31,6 +35,10 @@ "note": null, "command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia" }, + "cuda.z": { + "note": null, + "command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia" + }, "rocm5.x": { "note": "NOTE: Conda packages are not currently available for ROCm, please use pip instead
", "command": null @@ -62,6 +70,13 @@ "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip" } }, + "cuda.z": { + "note": null, + "versions": { + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-shared-with-deps-latest.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-cxx11-abi-shared-with-deps-latest.zip" + } + }, "rocm5.x": { "note": null, "versions": { @@ -83,6 +98,11 @@ "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "default": true }, + "cuda.z": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", + "default": true + }, "rocm5.x": { "note": "# ROCm is not available on MacOS, please use default package", "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", @@ -104,6 +124,11 @@ "command": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "default": true }, + "cuda.z": { + "note": "# CUDA is not available on MacOS, please use default package", + "command": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", + "default": true + }, "rocm5.x": { "note": "# ROCm is not available on MacOS, please use default package", "command": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", @@ -135,6 +160,13 @@ "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip" } }, + "cuda.z": { + "note": null, + "default": true, + "versions": { + "Download arm64 libtorch here (ROCm and CUDA are not supported):": "https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip" + } + }, "rocm5.x": { "note": null, "default": true, @@ -158,6 +190,10 @@ "note": null, "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121" }, + "cuda.z": { + "note": null, + "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124" + }, "rocm5.x": { "note": "NOTE: ROCm is not available on Windows", "command": null @@ -172,6 +208,10 @@ "note": null, "command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia" }, + "cuda.z": { + "note": null, + "command": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia" + }, "rocm5.x": { "note": "NOTE: ROCm is not available on Windows", "command": null @@ -203,6 +243,13 @@ "Download here (Debug version):": "https://download.pytorch.org/libtorch/nightly/cu121/libtorch-win-shared-with-deps-debug-latest.zip" } }, + "cuda.z": { + "note": null, + "versions": { + "Download here (Release version):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-win-shared-with-deps-latest.zip", + "Download here (Debug version):": "https://download.pytorch.org/libtorch/nightly/cu124/libtorch-win-shared-with-deps-debug-latest.zip" + } + }, "rocm5.x": { "note": "NOTE: ROCm is not available on Windows", "versions": null @@ -4335,4 +4382,4 @@ } } } -} \ No newline at end of file +} diff --git a/scripts/gen_quick_start_module.py b/scripts/gen_quick_start_module.py index 12a2c5862a51..a7ec1930c1a2 100755 --- a/scripts/gen_quick_start_module.py +++ b/scripts/gen_quick_start_module.py @@ -37,12 +37,14 @@ class OperatingSystem(Enum): "accnone": ("cpu", ""), "cuda.x": ("cuda", "11.8"), "cuda.y": ("cuda", "12.1"), + "cuda.z": ("cuda", "12.4"), "rocm5.x": ("rocm", "6.0") }, "release": { "accnone": ("cpu", ""), "cuda.x": ("cuda", "11.8"), "cuda.y": ("cuda", "12.1"), + "cuda.z": ("cuda", "12.4"), "rocm5.x": ("rocm", "6.0") } } @@ -184,16 +186,16 @@ def gen_ver_list(chan, gpu_arch_type): for chan in ("nightly", "release"): cuda_ver_list = gen_ver_list(chan, "cuda") rocm_ver_list = gen_ver_list(chan, "rocm") - cuda_list = sorted(cuda_ver_list.values())[-2:] + cuda_list = cuda_ver_list.values() acc_arch_ver_map[chan]["rocm5.x"] = ("rocm", max(rocm_ver_list.values())) - for cuda_ver, label in zip(cuda_list, ["cuda.x", "cuda.y"]): + for cuda_ver, label in zip(cuda_list, ["cuda.x", "cuda.y", "cuda.z"]): acc_arch_ver_map[chan][label] = ("cuda", cuda_ver) def main(): parser = argparse.ArgumentParser() parser.add_argument('--autogenerate', dest='autogenerate', action='https://github.com/store_true') - parser.set_defaults(autogenerate=False) + parser.set_defaults(autogenerate=True) options = parser.parse_args() versions = read_published_versions() @@ -213,9 +215,9 @@ def main(): template = read_quick_start_module_template() versions_str = json.dumps(gen_install_matrix(versions)) - template = template.replace("{{ installMatrix }}", versions_str) - template = template.replace("{{ VERSION }}", f"\"Stable ({versions['latest_stable']})\"") - print(template.replace("{{ ACC ARCH MAP }}", json.dumps(acc_arch_ver_map))) + #template = template.replace("{{ installMatrix }}", versions_str) + #template = template.replace("{{ VERSION }}", f"\"Stable ({versions['latest_stable']})\"") + #print(template.replace("{{ ACC ARCH MAP }}", json.dumps(acc_arch_ver_map))) if __name__ == "__main__": main() From 2aa6ae64f1614296d0db4389e1b5c644fa472757 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 09:57:05 -0700 Subject: [PATCH 02/13] test --- scripts/gen_quick_start_module.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/scripts/gen_quick_start_module.py b/scripts/gen_quick_start_module.py index a7ec1930c1a2..2b6914003983 100755 --- a/scripts/gen_quick_start_module.py +++ b/scripts/gen_quick_start_module.py @@ -186,7 +186,7 @@ def gen_ver_list(chan, gpu_arch_type): for chan in ("nightly", "release"): cuda_ver_list = gen_ver_list(chan, "cuda") rocm_ver_list = gen_ver_list(chan, "rocm") - cuda_list = cuda_ver_list.values() + cuda_list = sorted(cuda_ver_list.values()) acc_arch_ver_map[chan]["rocm5.x"] = ("rocm", max(rocm_ver_list.values())) for cuda_ver, label in zip(cuda_list, ["cuda.x", "cuda.y", "cuda.z"]): acc_arch_ver_map[chan][label] = ("cuda", cuda_ver) @@ -195,7 +195,7 @@ def gen_ver_list(chan, gpu_arch_type): def main(): parser = argparse.ArgumentParser() parser.add_argument('--autogenerate', dest='autogenerate', action='https://github.com/store_true') - parser.set_defaults(autogenerate=True) + parser.set_defaults(autogenerate=False) options = parser.parse_args() versions = read_published_versions() @@ -215,9 +215,9 @@ def main(): template = read_quick_start_module_template() versions_str = json.dumps(gen_install_matrix(versions)) - #template = template.replace("{{ installMatrix }}", versions_str) - #template = template.replace("{{ VERSION }}", f"\"Stable ({versions['latest_stable']})\"") - #print(template.replace("{{ ACC ARCH MAP }}", json.dumps(acc_arch_ver_map))) + template = template.replace("{{ installMatrix }}", versions_str) + template = template.replace("{{ VERSION }}", f"\"Stable ({versions['latest_stable']})\"") + print(template.replace("{{ ACC ARCH MAP }}", json.dumps(acc_arch_ver_map))) if __name__ == "__main__": main() From fdecf053f44f3fe929003e407b1fd3308d62d683 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 10:16:47 -0700 Subject: [PATCH 03/13] test --- _includes/quick_start_local.html | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/_includes/quick_start_local.html b/_includes/quick_start_local.html index 06ad1f0e7ce5..b40af32b57ee 100644 --- a/_includes/quick_start_local.html +++ b/_includes/quick_start_local.html @@ -87,19 +87,19 @@
Compute Platform
-
+
CUDA 11.8
-
+
CUDA 12.1
-
+
CUDA 12.4
-
+
ROCm 5.2
-
+
CPU
From b156d9421a50c2c2c0a9b3decf6c7298d815f057 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 10:54:45 -0700 Subject: [PATCH 04/13] test --- _includes/quick_start_local.html | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/_includes/quick_start_local.html b/_includes/quick_start_local.html index b40af32b57ee..d56c586a2f02 100644 --- a/_includes/quick_start_local.html +++ b/_includes/quick_start_local.html @@ -96,10 +96,10 @@
CUDA 12.4
-
+
ROCm 5.2
-
+
CPU
From 6973d8a2ef3ad5d62c536e9dc3ffa69430642b17 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 11:42:26 -0700 Subject: [PATCH 05/13] test --- _includes/quick-start-module.js | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/_includes/quick-start-module.js b/_includes/quick-start-module.js index 5cc118a3b280..424a88048168 100644 --- a/_includes/quick-start-module.js +++ b/_includes/quick-start-module.js @@ -129,7 +129,10 @@ function changeVersion(ptbuild) { for (const [arch_key, info] of archInfoMap) { var elems = document.querySelectorAll('[id^="'+arch_key+'"]'); for (var i=0; i < elems.length;i++) { - elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] + if(archMap[elems[i].id]) + elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] + else + elems[i].style.textDecoration = "line-through"; } } var stable_element = document.getElementById("stable"); From 18a166ec3226bbed91d5154998c633505c3bd785 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 11:50:58 -0700 Subject: [PATCH 06/13] add_cuda124 --- assets/quick-start-module.js | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index 8d90d1664ae4..3ee53bef1fbf 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -129,7 +129,10 @@ function changeVersion(ptbuild) { for (const [arch_key, info] of archInfoMap) { var elems = document.querySelectorAll('[id^="'+arch_key+'"]'); for (var i=0; i < elems.length;i++) { - elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] + if(archMap[elems[i].id]) + elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] + else + elems[i].style.textDecoration = "line-through"; } } var stable_element = document.getElementById("stable"); @@ -263,4 +266,3 @@ function commandMessage(key) { // Set cuda version right away changeVersion("stable") - From bf286530f1a878a4c0162a1d02c45aba0334f5e5 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 12:02:19 -0700 Subject: [PATCH 07/13] test --- _includes/quick-start-module.js | 18 ++++++++++++++++-- assets/quick-start-module.js | 18 ++++++++++++++++-- 2 files changed, 32 insertions(+), 4 deletions(-) diff --git a/_includes/quick-start-module.js b/_includes/quick-start-module.js index 424a88048168..3695b853935c 100644 --- a/_includes/quick-start-module.js +++ b/_includes/quick-start-module.js @@ -105,6 +105,12 @@ function getPreferredCuda(os) { // Disable compute platform not supported on OS function disableUnsupportedPlatforms(os) { + + if(opts.ptbuild == "preview") + archMap = version_map.nightly + else + archMap = version_map.release + for (const [arch_key, info] of archInfoMap) { var elems = document.querySelectorAll('[id^="'+arch_key+'"]'); if (elems == null) { @@ -114,6 +120,11 @@ function disableUnsupportedPlatforms(os) { for (var i=0; i < elems.length;i++) { var supported = info.platforms.has(os); elems[i].style.textDecoration = supported ? "" : "line-through"; + + // Officially supported arch but not available + if(!archMap[elems[i].id]) { + elems[i].style.textDecoration = "line-through"; + } } } } @@ -129,10 +140,13 @@ function changeVersion(ptbuild) { for (const [arch_key, info] of archInfoMap) { var elems = document.querySelectorAll('[id^="'+arch_key+'"]'); for (var i=0; i < elems.length;i++) { - if(archMap[elems[i].id]) + if(archMap[elems[i].id]) { + elems[i].style.textDecoration = ""; elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] - else + } + else { elems[i].style.textDecoration = "line-through"; + } } } var stable_element = document.getElementById("stable"); diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index 3ee53bef1fbf..5001e6ab8166 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -105,6 +105,12 @@ function getPreferredCuda(os) { // Disable compute platform not supported on OS function disableUnsupportedPlatforms(os) { + + if(opts.ptbuild == "preview") + archMap = version_map.nightly + else + archMap = version_map.release + for (const [arch_key, info] of archInfoMap) { var elems = document.querySelectorAll('[id^="'+arch_key+'"]'); if (elems == null) { @@ -114,6 +120,11 @@ function disableUnsupportedPlatforms(os) { for (var i=0; i < elems.length;i++) { var supported = info.platforms.has(os); elems[i].style.textDecoration = supported ? "" : "line-through"; + + // Officially supported arch but not available + if(!archMap[elems[i].id]) { + elems[i].style.textDecoration = "line-through"; + } } } } @@ -129,10 +140,13 @@ function changeVersion(ptbuild) { for (const [arch_key, info] of archInfoMap) { var elems = document.querySelectorAll('[id^="'+arch_key+'"]'); for (var i=0; i < elems.length;i++) { - if(archMap[elems[i].id]) + if(archMap[elems[i].id]) { + elems[i].style.textDecoration = ""; elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] - else + } + else { elems[i].style.textDecoration = "line-through"; + } } } var stable_element = document.getElementById("stable"); From 3976feb1f759e9e56aeff2063355bcef9ff06b5e Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 12:13:21 -0700 Subject: [PATCH 08/13] test --- assets/quick-start-module.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index 5001e6ab8166..f834205b7535 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -11,7 +11,7 @@ var archInfoMap = new Map([ ['accnone', {title: "CPU", platforms: new Set(['linux', 'macos', 'windows'])}] ]); -let version_map={"nightly": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"]}} +let version_map={"nightly": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.0"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"]}} let stable_version="Stable (2.3.0)"; var default_selected_os = getAnchorSelectedOS() || getDefaultSelectedOS(); From 26ad83f2de59d2f90c6d98f79f14cd43d4c135d6 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 12:34:51 -0700 Subject: [PATCH 09/13] Generate --- published_versions.json | 8 ++++---- scripts/gen_quick_start_module.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/published_versions.json b/published_versions.json index 450967483a3c..3fe293299e68 100644 --- a/published_versions.json +++ b/published_versions.json @@ -23,7 +23,7 @@ }, "rocm5.x": { "note": null, - "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0" + "command": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1" } }, "conda": { @@ -80,8 +80,8 @@ "rocm5.x": { "note": null, "versions": { - "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/rocm6.0/libtorch-shared-with-deps-latest.zip", - "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/rocm6.0/libtorch-cxx11-abi-shared-with-deps-latest.zip" + "Download here (Pre-cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/rocm6.1/libtorch-shared-with-deps-latest.zip", + "Download here (cxx11 ABI):": "https://download.pytorch.org/libtorch/nightly/rocm6.1/libtorch-cxx11-abi-shared-with-deps-latest.zip" } } } @@ -4382,4 +4382,4 @@ } } } -} +} \ No newline at end of file diff --git a/scripts/gen_quick_start_module.py b/scripts/gen_quick_start_module.py index 2b6914003983..a988233a1e51 100755 --- a/scripts/gen_quick_start_module.py +++ b/scripts/gen_quick_start_module.py @@ -195,7 +195,7 @@ def gen_ver_list(chan, gpu_arch_type): def main(): parser = argparse.ArgumentParser() parser.add_argument('--autogenerate', dest='autogenerate', action='https://github.com/store_true') - parser.set_defaults(autogenerate=False) + parser.set_defaults(autogenerate=True) options = parser.parse_args() versions = read_published_versions() From 994787a723316214eec32ceb3f7d0bc605f4fad6 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 12:40:49 -0700 Subject: [PATCH 10/13] test --- assets/quick-start-module.js | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index f834205b7535..ecf816e56b4f 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -11,7 +11,7 @@ var archInfoMap = new Map([ ['accnone', {title: "CPU", platforms: new Set(['linux', 'macos', 'windows'])}] ]); -let version_map={"nightly": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.0"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"]}} +let version_map={"nightly": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.1"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.0"]}} let stable_version="Stable (2.3.0)"; var default_selected_os = getAnchorSelectedOS() || getDefaultSelectedOS(); @@ -265,7 +265,7 @@ $("[data-toggle='cloud-dropdown']").on("click", function(e) { }); function commandMessage(key) { - var object = {"preview,pip,linux,accnone,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,linux,cuda.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "preview,pip,linux,cuda.y,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121", "preview,pip,linux,rocm5.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0", "preview,conda,linux,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia", "preview,conda,linux,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia", "preview,conda,linux,rocm5.x,python": "NOTE: Conda packages are not currently available for ROCm, please use pip instead
", "preview,conda,linux,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly", "preview,libtorch,linux,accnone,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,cuda.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,cuda.y,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,rocm5.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/rocm6.0/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/rocm6.0/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,pip,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,accnone,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,conda,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,accnone,python": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,libtorch,macos,accnone,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,cuda.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,cuda.y,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,rocm5.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,pip,windows,accnone,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,windows,cuda.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "preview,pip,windows,cuda.y,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121", "preview,pip,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "preview,conda,windows,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia", "preview,conda,windows,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia", "preview,conda,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "preview,conda,windows,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly", "preview,libtorch,windows,accnone,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,cuda.x,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,cuda.y,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,rocm5.x,cplusplus": "NOTE: ROCm is not available on Windows", "stable,pip,linux,accnone,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu", "stable,pip,linux,cuda.x,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118", "stable,pip,linux,cuda.y,python": "pip3 install torch torchvision torchaudio", "stable,pip,linux,rocm5.x,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0", "stable,conda,linux,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia", "stable,conda,linux,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia", "stable,conda,linux,rocm5.x,python": "NOTE: Conda packages are not currently available for ROCm, please use pip instead
", "stable,conda,linux,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch", "stable,libtorch,linux,accnone,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.3.0%2Bcpu.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcpu.zip", "stable,libtorch,linux,cuda.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cu118/libtorch-shared-with-deps-2.3.0%2Bcu118.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcu118.zip", "stable,libtorch,linux,cuda.y,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.3.0%2Bcu121.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcu121.zip", "stable,libtorch,linux,rocm5.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/rocm6.0/libtorch-shared-with-deps-2.3.0%2Brocm6.0.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/rocm6.0/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Brocm6.0.zip", "stable,pip,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
pip3 install torch torchvision torchaudio", "stable,pip,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
pip3 install torch torchvision torchaudio", "stable,pip,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
pip3 install torch torchvision torchaudio", "stable,pip,macos,accnone,python": "pip3 install torch torchvision torchaudio", "stable,conda,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,conda,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,conda,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,conda,macos,accnone,python": "conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,libtorch,macos,accnone,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,libtorch,macos,cuda.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,libtorch,macos,cuda.y,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,libtorch,macos,rocm5.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,pip,windows,accnone,python": "pip3 install torch torchvision torchaudio", "stable,pip,windows,cuda.x,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118", "stable,pip,windows,cuda.y,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121", "stable,pip,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "stable,conda,windows,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia", "stable,conda,windows,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia", "stable,conda,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "stable,conda,windows,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch", "stable,libtorch,windows,accnone,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-2.3.0%2Bcpu.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-2.3.0%2Bcpu.zip", "stable,libtorch,windows,cuda.x,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-2.3.0%2Bcu118.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-debug-2.3.0%2Bcu118.zip", "stable,libtorch,windows,cuda.y,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-2.3.0%2Bcu121.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-debug-2.3.0%2Bcu121.zip", "stable,libtorch,windows,rocm5.x,cplusplus": "NOTE: ROCm is not available on Windows"}; + var object = {"preview,pip,linux,accnone,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,linux,cuda.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "preview,pip,linux,cuda.y,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121", "preview,pip,linux,cuda.z,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "preview,pip,linux,rocm5.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1", "preview,conda,linux,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia", "preview,conda,linux,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia", "preview,conda,linux,cuda.z,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia", "preview,conda,linux,rocm5.x,python": "NOTE: Conda packages are not currently available for ROCm, please use pip instead
", "preview,conda,linux,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly", "preview,libtorch,linux,accnone,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,cuda.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,cuda.y,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,cuda.z,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu124/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cu124/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,libtorch,linux,rocm5.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/rocm6.1/libtorch-shared-with-deps-latest.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/rocm6.1/libtorch-cxx11-abi-shared-with-deps-latest.zip", "preview,pip,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,cuda.z,python": "# CUDA is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,macos,accnone,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,conda,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,cuda.z,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,conda,macos,accnone,python": "conda install pytorch-nightly::pytorch torchvision torchaudio -c pytorch-nightly", "preview,libtorch,macos,accnone,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,cuda.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,cuda.y,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,cuda.z,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,libtorch,macos,rocm5.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-macos-arm64-latest.zip", "preview,pip,windows,accnone,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", "preview,pip,windows,cuda.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu118", "preview,pip,windows,cuda.y,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121", "preview,pip,windows,cuda.z,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124", "preview,pip,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "preview,conda,windows,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch-nightly -c nvidia", "preview,conda,windows,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-nightly -c nvidia", "preview,conda,windows,cuda.z,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch-nightly -c nvidia", "preview,conda,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "preview,conda,windows,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly", "preview,libtorch,windows,accnone,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,cuda.x,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cu118/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,cuda.y,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cu121/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,cuda.z,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/nightly/cu124/libtorch-win-shared-with-deps-latest.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/nightly/cu124/libtorch-win-shared-with-deps-debug-latest.zip", "preview,libtorch,windows,rocm5.x,cplusplus": "NOTE: ROCm is not available on Windows", "stable,pip,linux,accnone,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu", "stable,pip,linux,cuda.x,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118", "stable,pip,linux,cuda.y,python": "pip3 install torch torchvision torchaudio", "stable,pip,linux,rocm5.x,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0", "stable,conda,linux,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia", "stable,conda,linux,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia", "stable,conda,linux,rocm5.x,python": "NOTE: Conda packages are not currently available for ROCm, please use pip instead
", "stable,conda,linux,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch", "stable,libtorch,linux,accnone,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.3.0%2Bcpu.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcpu.zip", "stable,libtorch,linux,cuda.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cu118/libtorch-shared-with-deps-2.3.0%2Bcu118.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cu118/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcu118.zip", "stable,libtorch,linux,cuda.y,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.3.0%2Bcu121.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Bcu121.zip", "stable,libtorch,linux,rocm5.x,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/rocm6.0/libtorch-shared-with-deps-2.3.0%2Brocm6.0.zip
Download here (cxx11 ABI):
https://download.pytorch.org/libtorch/rocm6.0/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Brocm6.0.zip", "stable,pip,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
pip3 install torch torchvision torchaudio", "stable,pip,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
pip3 install torch torchvision torchaudio", "stable,pip,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
pip3 install torch torchvision torchaudio", "stable,pip,macos,accnone,python": "pip3 install torch torchvision torchaudio", "stable,conda,macos,cuda.x,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,conda,macos,cuda.y,python": "# CUDA is not available on MacOS, please use default package
conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,conda,macos,rocm5.x,python": "# ROCm is not available on MacOS, please use default package
conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,conda,macos,accnone,python": "conda install pytorch::pytorch torchvision torchaudio -c pytorch", "stable,libtorch,macos,accnone,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,libtorch,macos,cuda.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,libtorch,macos,cuda.y,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,libtorch,macos,rocm5.x,cplusplus": "Download arm64 libtorch here (ROCm and CUDA are not supported):
https://download.pytorch.org/libtorch/cpu/libtorch-macos-arm64-2.3.0.zip", "stable,pip,windows,accnone,python": "pip3 install torch torchvision torchaudio", "stable,pip,windows,cuda.x,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118", "stable,pip,windows,cuda.y,python": "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121", "stable,pip,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "stable,conda,windows,cuda.x,python": "conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia", "stable,conda,windows,cuda.y,python": "conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia", "stable,conda,windows,rocm5.x,python": "NOTE: ROCm is not available on Windows", "stable,conda,windows,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch", "stable,libtorch,windows,accnone,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-2.3.0%2Bcpu.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-debug-2.3.0%2Bcpu.zip", "stable,libtorch,windows,cuda.x,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-2.3.0%2Bcu118.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/cu118/libtorch-win-shared-with-deps-debug-2.3.0%2Bcu118.zip", "stable,libtorch,windows,cuda.y,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-2.3.0%2Bcu121.zip
Download here (Debug version):
https://download.pytorch.org/libtorch/cu121/libtorch-win-shared-with-deps-debug-2.3.0%2Bcu121.zip", "stable,libtorch,windows,rocm5.x,cplusplus": "NOTE: ROCm is not available on Windows"}; if (!object.hasOwnProperty(key)) { $("#command").html( @@ -280,3 +280,4 @@ function commandMessage(key) { // Set cuda version right away changeVersion("stable") + From 7f3a9c9c9fb56fcbf1891092362d6071b0c384b2 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 21 May 2024 12:57:04 -0700 Subject: [PATCH 11/13] test --- assets/quick-start-module.js | 2 +- scripts/gen_quick_start_module.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index ecf816e56b4f..4fce6ecf5b22 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -11,7 +11,7 @@ var archInfoMap = new Map([ ['accnone', {title: "CPU", platforms: new Set(['linux', 'macos', 'windows'])}] ]); -let version_map={"nightly": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.1"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.0"]}} +let version_map={"nightly": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "cuda.z": ["cuda", "12.4"], "rocm5.x": ["rocm", "6.1"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"]}} let stable_version="Stable (2.3.0)"; var default_selected_os = getAnchorSelectedOS() || getDefaultSelectedOS(); diff --git a/scripts/gen_quick_start_module.py b/scripts/gen_quick_start_module.py index a988233a1e51..2788503abf16 100755 --- a/scripts/gen_quick_start_module.py +++ b/scripts/gen_quick_start_module.py @@ -44,7 +44,6 @@ class OperatingSystem(Enum): "accnone": ("cpu", ""), "cuda.x": ("cuda", "11.8"), "cuda.y": ("cuda", "12.1"), - "cuda.z": ("cuda", "12.4"), "rocm5.x": ("rocm", "6.0") } } From fd409d3e016c8182975b4d6320b539d2c2cfb4ff Mon Sep 17 00:00:00 2001 From: atalman Date: Wed, 22 May 2024 07:19:19 -0700 Subject: [PATCH 12/13] test --- _includes/quick-start-module.js | 5 +++-- assets/quick-start-module.js | 3 ++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/_includes/quick-start-module.js b/_includes/quick-start-module.js index 3695b853935c..37da3ab79893 100644 --- a/_includes/quick-start-module.js +++ b/_includes/quick-start-module.js @@ -143,8 +143,7 @@ function changeVersion(ptbuild) { if(archMap[elems[i].id]) { elems[i].style.textDecoration = ""; elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] - } - else { + } else { elems[i].style.textDecoration = "line-through"; } } @@ -207,6 +206,8 @@ function selectedOption(option, selection, category) { } } else if (category == "ptbuild") { changeVersion(opts.ptbuild); + //make sure unsupported platforms are disabled + disableUnsupportedPlatforms(opts.os); } commandMessage(buildMatcher()); if (category === "os") { diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index 4fce6ecf5b22..4d07c46d80bc 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -207,6 +207,8 @@ function selectedOption(option, selection, category) { } } else if (category == "ptbuild") { changeVersion(opts.ptbuild); + //make sure unsupported platforms are disabled + disableUnsupportedPlatforms(opts.os); } commandMessage(buildMatcher()); if (category === "os") { @@ -280,4 +282,3 @@ function commandMessage(key) { // Set cuda version right away changeVersion("stable") - From 95f748354f0e7621c9c75cac64d516b9f44e44f3 Mon Sep 17 00:00:00 2001 From: atalman Date: Wed, 22 May 2024 07:20:13 -0700 Subject: [PATCH 13/13] test --- assets/quick-start-module.js | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js index 4d07c46d80bc..842721655ffc 100644 --- a/assets/quick-start-module.js +++ b/assets/quick-start-module.js @@ -143,8 +143,7 @@ function changeVersion(ptbuild) { if(archMap[elems[i].id]) { elems[i].style.textDecoration = ""; elems[i].children[0].textContent = info.title + " " + archMap[elems[i].id][1] - } - else { + } else { elems[i].style.textDecoration = "line-through"; } } @@ -282,3 +281,4 @@ function commandMessage(key) { // Set cuda version right away changeVersion("stable") +