diff --git a/_sass/quick-start-module.scss b/_sass/quick-start-module.scss
index 06d5500bd61f..b510c02c09b9 100644
--- a/_sass/quick-start-module.scss
+++ b/_sass/quick-start-module.scss
@@ -135,6 +135,31 @@
left: 0;
}
+ .row.cuda {
+ justify-content: space-evenly;
+ }
+
+ .w20 {
+ padding: 0 15px;
+ flex: 1;
+ }
+
+ .flex1{
+ flex: 2;
+ }
+
+ .acc-arrow {
+ float: right;
+ position: absolute;
+ margin-top: 8px;
+ margin-left: 5px;
+ }
+
+ .resources-dropdown-menu {
+ width: 300px;
+ left: 400px;
+ }
+
#command {
color: #4a4a4a;
background-color: $white;
@@ -341,7 +366,7 @@
color: $not_quite_black;
#microsoft-azure {
- p{
+ p {
color: $not_quite_black;
margin: 0;
padding: 0;
@@ -370,7 +395,7 @@
li {
margin-top: 0;
- position:relative;
+ position: relative;
padding-left: rem(80px);
@include small-desktop {
diff --git a/assets/quick-start-module.js b/assets/quick-start-module.js
index 8d90d1664ae4..7c6c99c20ba5 100644
--- a/assets/quick-start-module.js
+++ b/assets/quick-start-module.js
@@ -6,12 +6,15 @@ var supportedOperatingSystems = new Map([
]);
var archInfoMap = new Map([
- ['cuda', {title: "CUDA", platforms: new Set(['linux', 'windows'])}],
- ['rocm', {title: "ROCm", platforms: new Set(['linux'])}],
- ['accnone', {title: "CPU", platforms: new Set(['linux', 'macos', 'windows'])}]
+ ['cuda', {title: "CUDA", platforms: new Set(['linux', 'windows']) }],
+ ['rocm', {title: "ROCm", platforms: new Set(['linux']) }],
+ ['accnone', {title: "CPU", platforms: new Set(['linux', 'macos', 'windows'])}],
+ ['huaweiascend', {title: "Huawei Ascend", platforms: new Set(['linux'])}],
+ ['intelgaudi', {title: "Intel Gaudi", platforms: new Set(['linux'])}],
+ ['intelextensionforpytorch', {title: "Intel Extension for PyTorch", platforms: new Set(['linux'])}],
]);
-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"], "rocm5.x": ["rocm", "6.0"]}, "release": {"accnone": ["cpu", ""], "cuda.x": ["cuda", "11.8"], "cuda.y": ["cuda", "12.1"], "rocm5.x": ["rocm", "6.0"], "huaweiascend": ["huaweiascend", ""], "intelgaudi": ["intelgaudi", ""], "intelextensionforpytorch": ["intelextensionforpytorch", ""]}}
let stable_version="Stable (2.3.0)";
var default_selected_os = getAnchorSelectedOS() || getDefaultSelectedOS();
@@ -34,6 +37,7 @@ var package = $(".package > .option");
var language = $(".language > .option");
var cuda = $(".cuda > .option");
var ptbuild = $(".ptbuild > .option");
+var accelerator = null;
os.on("click", function() {
selectedOption(os, this, "os");
@@ -45,7 +49,10 @@ language.on("click", function() {
selectedOption(language, this, "language");
});
cuda.on("click", function() {
- selectedOption(cuda, this, "cuda");
+ if(this.id != 'accelerators') {
+ accelerator = null;
+ selectedOption(cuda, this, "cuda");
+ }
});
ptbuild.on("click", function() {
selectedOption(ptbuild, this, "ptbuild")
@@ -63,7 +70,6 @@ $(function() {
}
});
-
// determine os (mac, linux, windows) based on user's platform
function getDefaultSelectedOS() {
var platform = navigator.platform.toLowerCase();
@@ -116,11 +122,27 @@ function disableUnsupportedPlatforms(os) {
elems[i].style.textDecoration = supported ? "" : "line-through";
}
}
+ if (accelerator) {
+ disableUnsupportedAccelerator(os);
+ }
+}
+
+// Disable accelerator not supported on OS
+function disableUnsupportedAccelerator(os) {
+ let supported = archInfoMap.get(accelerator.toLowerCase().replaceAll(" ", "")).platforms.has(os);
+ document.getElementById("accelerators").style.textDecoration = supported ? "" : "line-through";
+}
+
+// Choose accelerator
+function chooseAccelerator(acceleratorName){
+ accelerator = acceleratorName;
+ document.getElementById("accelerator").innerHTML = accelerator;
+ selectedOption(cuda, this, "cuda");
+ disableUnsupportedAccelerator(opts.os);
}
// Change compute versions depending on build type
function changeVersion(ptbuild) {
-
if(ptbuild == "preview")
archMap = version_map.nightly
else
@@ -136,8 +158,6 @@ function changeVersion(ptbuild) {
stable_element.children[0].textContent = stable_version;
}
-
-
// Change accnone name depending on OS type
function changeAccNoneName(osname) {
var accnone_element = document.getElementById("accnone");
@@ -155,6 +175,11 @@ function changeAccNoneName(osname) {
function selectedOption(option, selection, category) {
$(option).removeClass("selected");
$(selection).addClass("selected");
+ if (!accelerator) {
+ $('#accelerators').removeClass("selected");
+ } else {
+ $('#accelerators').addClass("selected");
+ }
opts[category] = selection.id;
if (category === "pm") {
var elements = document.getElementsByClassName("language")[0].children;
@@ -223,7 +248,7 @@ function buildMatcher() {
"," +
opts.os.toLowerCase() +
"," +
- opts.cuda.toLowerCase() +
+ (accelerator ? accelerator.toLowerCase().replaceAll(' ', '') : opts.cuda.toLowerCase()) +
"," +
opts.language.toLowerCase()
);
@@ -248,7 +273,117 @@ $("[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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload 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,rocm5.x,python": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0",
+ "preview,pip,linux,intelgaudi,python": "
NOTE: Preview (Nightly) is not currently available for Intel Gaudi, please use stable instead
",
+ "preview,pip,linux,intelextensionforpytorch,python": "
NOTE: Preview (Nightly) is not currently available for Intel Extension for PyTorch, please use stable instead
",
+ "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,conda,linux,intelgaudi,python": "
NOTE: Conda packages are not currently available for Intel Gaudi, please use pip instead
",
+ "preview,conda,linux,intelextensionforpytorch,python": "
NOTE: Conda packages are not currently available for Intel Extension for PyTorch, please use pip instead
",
+ "preview,libtorch,linux,accnone,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload here (cxx11 ABI):
https://download.pytorch.org/libtorch/nightly/rocm6.0/libtorch-cxx11-abi-shared-with-deps-latest.zip",
+ "preview,libtorch,linux,intelgaudi,cplusplus": "
NOTE: LibTorch is currently available for Intel Gaudi, please use pip instead
",
+ "preview,libtorch,linux,intelextensionforpytorch,cplusplus": "
NOTE: LibTorch is currently available for Intel Extension for PyTorch, please use pip instead
",
+ "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,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,pip,macos,intelgaudi,python": "
NOTE: Intel Gaudi is not available on MacOS",
+ "preview,pip,macos,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on MacOS",
+ "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,conda,macos,intelgaudi,python": "
NOTE: Intel Gaudi is not available on MacOS",
+ "preview,conda,macos,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on MacOS",
+ "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,libtorch,macos,intelgaudi,python": "
NOTE: Intel Gaudi is not available on MacOS",
+ "preview,libtorch,macos,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on MacOS",
+ "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,pip,windows,intelgaudi,python": "
NOTE: Intel Gaudi is not available on Windows",
+ "preview,pip,windows,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch 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,intelgaudi,python": "
NOTE: Intel Gaudi is not available on Windows",
+ "preview,conda,windows,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch 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.zipDownload 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.zipDownload 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.zipDownload 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",
+ "preview,libtorch,windows,intelgaudi,cplusplus": "
NOTE: Intel Gaudi is not available on Windows",
+ "preview,libtorch,windows,intelextensionforpytorch,cplusplus": "
NOTE: Intel Extension for PyTorch 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,pip,linux,intelgaudi,python": "
NOTE: Newest PyTorch version is not yet supported. For installation commands please refer to
previous versions of PyTorch and
Intel Gaudi documentation.",
+ "stable,pip,linux,intelextensionforpytorch,python": "# CPU
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip3 install intel-extension-for-pytorch
# GPU
NOTE: Newest PyTorch version is not yet supported.
For installation command please refer to
previous versions of PyTorch.",
+ "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,conda,linux,intelgaudi,python": "
NOTE: Conda packages are not currently available for Intel Gaudi, please use pip instead
",
+ "stable,conda,linux,intelextensionforpytorch,python": "
NOTE: Conda packages are not currently available for Intel Extension for PyTorch, please use pip instead
",
+ "stable,libtorch,linux,accnone,cplusplus": "Download here (Pre-cxx11 ABI):
https://download.pytorch.org/libtorch/cpu/libtorch-shared-with-deps-2.3.0%2Bcpu.zipDownload 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.zipDownload 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.zipDownload 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.zipDownload here (cxx11 ABI):
https://download.pytorch.org/libtorch/rocm6.0/libtorch-cxx11-abi-shared-with-deps-2.3.0%2Brocm6.0.zip",
+ "stable,libtorch,linux,intelgaudi,cplusplus": "
NOTE: LibTorch is currently available for Intel Gaudi, please use pip instead
",
+ "stable,libtorch,linux,intelextensionforpytorch,cplusplus": "
NOTE: LibTorch is currently available for Intel Extension for PyTorch, please use pip instead
",
+ "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,pip,macos,intelgaudi,python": "
NOTE: Intel Gaudi is not available on MacOS",
+ "stable,pip,macos,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on MacOS",
+ "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,intelgaudi,python": "
NOTE: Intel Gaudi is not available on MacOS",
+ "stable,conda,macos,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on MacOS",
+ "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,libtorch,macos,intelgaudi,cplusplus": "
NOTE: Intel Gaudi is not available on MacOS",
+ "stable,libtorch,macos,intelextensionforpytorch,cplusplus": "
NOTE: Intel Extension for PyTorch is not available on MacOS",
+ "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,pip,windows,intelgaudi,python": "
NOTE: Intel Gaudi is not available on Windows",
+ "stable,pip,windows,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on Windows",
+ "stable,conda,windows,accnone,python": "conda install pytorch torchvision torchaudio cpuonly -c pytorch",
+ "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,intelgaudi,python": "
NOTE: Intel Gaudi is not available on Windows",
+ "stable,conda,windows,intelextensionforpytorch,python": "
NOTE: Intel Extension for PyTorch is not available on Windows",
+ "stable,libtorch,windows,accnone,cplusplus": "Download here (Release version):
https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-2.3.0%2Bcpu.zipDownload 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.zipDownload 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.zipDownload 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",
+ "stable,libtorch,windows,intelgaudi,cplusplus": "
NOTE: Intel Gaudi is not available on Windows",
+ "stable,libtorch,windows,intelextensionforpytorch,cplusplus": "
NOTE: Intel Extension for PyTorch is not available on Windows"
+ }
if (!object.hasOwnProperty(key)) {
$("#command").html(
+
+