saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Website] Update training install matrix (#12621)

Browse source 
* update training install matrix

* hide architecture option
This commit is contained in:
Faith Xu 2022-08-17 09:15:21 -07:00 committed by GitHub
parent 746130073b
commit f4d426291e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 84 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
index.html
View file

@ -305,14 +305,11 @@
<div class="row r-wrap mb-1 mb-md-0 mr-0">
<div class="col-md-3 r-heading">
<h3 id="ot_selectOS">Platform</h3>
<p id="ot_decriptionOS" class="sr-only">Platform list contains three items</p>
<p id="ot_decriptionOS" class="sr-only">Platform list contains one item</p>
</div>
<div class="col-md-9 r-content pr-0 pl-md-4" role="listbox" id="ot_listbox-1" aria-labelledby="ot_selectOS" aria-describedby="ot_decriptionOS">
<div class="row ot_os">
<div class="col r-option selected" role="option" tabindex="0" aria-selected="true" id="ot_linux"><span>Linux</span></div>
<div class="col r-option" role="option" tabindex="-1" aria-selected="false" id="ot_windows"><span>Windows</span></div>
<div class="col r-option" role="option" tabindex="-1" aria-selected="false" id="ot_mac"><span>Mac</span></div>
</div>
</div>
</div>
@ -320,43 +317,25 @@
<div class="row r-wrap mb-1 mb-md-0 mr-0">
<div class="col-md-3 r-heading">
<h3 id="ot_selectLanguage">API</h3>
<p id="ot_decriptionLanguage" class="sr-only">API list contains three items</p>
<p id="ot_decriptionLanguage" class="sr-only">API list contains one item</p>
</div>
<div class="col-md-9 r-content pr-0 pl-md-4" role="listbox" id="ot_listbox-2" aria-labelledby="ot_selectLanguage" aria-describedby="ot_decriptionLanguage">
<div class="row ot_language">
<div class="col-lg-2dot4 col r-option" role="option" tabindex="-1" aria-selected="false" id="ot_PyTorch18"><span>PyTorch 1.8.1</span></div>
<div class="col-lg-2dot4 col r-option selected" role="option" tabindex="0" aria-selected="true" id="ot_PyTorch19"><span>PyTorch 1.9</span></div>
<div class="col-lg-2dot4 col r-option" role="option" tabindex="-1" aria-selected="false" id="ot_C++"><span>C++</span></div>
</div>
</div>
</div>
<div class="row r-wrap mb-1 mb-md-0 mr-0">
<div class="col-md-3 r-heading ">
<h3 id="ot_selectArchitecture" class="align-self-center">Architecture</h3>
<p id="ot_decriptionArchitecture" class="sr-only">Architecture list contains one item</p>
</div>
<div class="col-md-9 r-content pr-0 pl-md-4" role="listbox" id="ot_listbox-3" aria-labelledby="ot_selectArchitecture" aria-describedby="ot_decriptionArchitecture">
<div class="row ot_architecture">
<div class="col r-option selected" role="option" tabindex="0" aria-selected="true" id="ot_X64">
<span><abbr>X64</abbr></span></div>
<div class="col r-option selected" role="option" tabindex="0" aria-selected="true" id="ot_ORTModule"><span>ORTModule for Pytorch</span></div>
</div>
</div>
</div>
<div class="row r-wrap mb-1 mb-md-0 mr-0">
<div class="col-md-3 r-heading">
<h3 id="ot_selectHardwareAcceleration">Hardware Acceleration</h3>
<p id="ot_decriptionHardwareAcceleration" class="sr-only">Hardware Acceleration list contains four items</p>
<p id="ot_decriptionHardwareAcceleration" class="sr-only">Hardware Acceleration list contains two items</p>
</div>
<div class="col-md-9 r-content pr-0 pl-md-4" role="listbox" id="ot_listbox-4" aria-labelledby="ot_selectHardwareAcceleration" aria-describedby="ot_decriptionHardwareAcceleration">
<div class="row ot_hardwareAcceleration">
<div class="col r-option version" role="option" tabindex="-1" aria-selected="false" id="ot_DefaultCPU"><span><abbr>Default CPU</abbr></span></div>
<div class="col r-option version selected" role="option" tabindex="0" aria-selected="true" id="ot_CUDA10"><span><abbr>CUDA 10.2</abbr></span></div>
<div class="col r-option version" role="option" tabindex="-1" aria-selected="false" id="ot_CUDA11"><span><abbr>CUDA 11.1</abbr></span></div>
<div class="col r-option version" role="option" tabindex="-1" aria-selected="false" id="ot_ROCm42"><span><abbr>ROCm 4.2 (Preview)</abbr></span></div>
<div class="col r-option version" role="option" tabindex="-1" aria-selected="false" id="ot_ROCm431"><span><abbr>ROCm 4.3.1 (Preview)</abbr></span></div>
<div class="col r-option version" role="option" tabindex="-1" aria-selected="false" id="ot_DNNL"><span><abbr>oneDNN</abbr></span></div>
<div class="col r-option version selected" role="option" tabindex="0" aria-selected="true" id="ot_CUDA"><span><abbr>CUDA</abbr></span></div>
<div class="col r-option version" role="option" tabindex="-1" aria-selected="false" id="ot_ROCm"><span><abbr>ROCm</abbr></span></div>
</div>
</div>
</div>

63
js/script.js
View file

@ -21,8 +21,8 @@ var ot_opts = {
// os: getAnchorSelectedOS() || getDefaultSelectedOS(),
ot_os: 'ot_linux',
ot_architecture: 'ot_X64',
ot_language: 'ot_PyTorch19',
ot_hardwareAcceleration: 'ot_CUDA10',
ot_language: 'ot_ORTModule',
ot_hardwareAcceleration: 'ot_CUDA',
};
var os = $(".os > .r-option");
@ -534,62 +534,13 @@ function ot_buildMatcher() {
}
var ot_validCombos = {
//linux
"ot_linux,ot_PyTorch18,ot_X64,ot_CUDA10":
"pip install torch-ort<br/> python -m torch_ort.configure",
"ot_linux,ot_PyTorch18,ot_X64,ot_CUDA11":
"pip install torch-ort -f https://download.onnxruntime.ai/onnxruntime_stable_cu111.html<br/> python -m torch_ort.configure",
"ot_linux,ot_ORTModule,ot_X64,ot_CUDA":
"pip install onnxruntime-training -f https://downloads.onnxruntime.ai/onnxruntime_stable_<b>&lt;cu_version*</b>&gt;.html<br/>pip install torch-ort<br/>python -m torch_ort.configure<br/><br/>*</b><a href='https://download.onnxruntime.ai/' target='blank'>Available versions</a>",
"ot_linux,ot_PyTorch18,ot_X64,ot_DefaultCPU":
"Follow sample notebook from <a href='https://github.com/microsoft/onnxruntime-training-examples' target='_blank'>here</a>",
"ot_linux,ot_PyTorch18,ot_X64,ot_ROCm42":
"This combination requires an AMD docker image. See <a href=https://github.com/pytorch/ort>the torch-ort repo</a> for details",
"ot_linux,ot_PyTorch18,ot_X64,ot_ROCm431":
"This combination requires an AMD docker image. See <a href=https://github.com/pytorch/ort>the torch-ort repo</a> for details",
"ot_linux,ot_PyTorch18,ot_X64,ot_DNNL":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_linux,ot_PyTorch19,ot_X64,ot_CUDA10":
"pip install torch-ort<br/> python -m torch_ort.configure",
"ot_linux,ot_PyTorch19,ot_X64,ot_CUDA11":
"pip install torch-ort -f https://download.onnxruntime.ai/onnxruntime_stable_cu111.html<br/> python -m torch_ort.configure",
"ot_linux,ot_PyTorch19,ot_X64,ot_ROCm42":
"(Preview) pip install torch -f https://download.pytorch.org/whl/rocm4.2/torch_stable.html<br/>pip install torch-ort -f https://download.onnxruntime.ai/onnxruntime_stable_rocm42.html<br/>python -m torch_ort.configure</br>",
"ot_linux,ot_PyTorch19,ot_X64,ot_ROCm431":
"This combination requires an AMD docker image. See <a href=https://github.com/pytorch/ort>the torch-ort repo</a> for details",
"ot_linux,ot_C++,ot_X64,ot_CUDA10":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_linux,ot_C++,ot_X64,ot_DefaultCPU":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_linux,ot_ORTModule,ot_X64,ot_ROCm":
"pip install onnxruntime-training -f https://downloads.onnxruntime.ai/onnxruntime_stable_<b>&lt;rocm_version*</b>&gt;.html<br/>pip install torch-ort<br/>python -m torch_ort.configure<br/><br/>*<a href='https://download.onnxruntime.ai/' target='blank'>Available versions</a>"
//windows
"ot_windows,ot_PyTorch18,ot_X64,ot_CUDA10":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_windows,ot_PyTorch18,ot_X64,ot_DefaultCPU":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_windows,ot_C++,ot_X64,ot_DefaultCPU":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_windows,ot_C++,ot_X64,ot_CUDA10":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
//mac
"ot_mac,ot_PyTorch18,ot_X64,ot_DefaultCPU":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>.",
"ot_mac,ot_C++,ot_X64,ot_DefaultCPU":
"This combination of resources is not fully tested. It may be possible to&nbsp;<a href='https://www.onnxruntime.ai/docs/build/training' target='_blank'>build from source</a>."
};
function ot_commandMessage(key) {
@ -617,7 +568,7 @@ function ot_commandMessage(key) {
// //console.log('key- '+key);
// var ot_object = {
// "ot_linux,ot_PyTorch18,ot_X64,ot_CUDA10":
// "ot_linux,ot_ORTModule,ot_X64,ot_CUDA10":
// "Follow sample notebook from <a href='https://github.com/microsoft/onnxruntime-training-examples' target='_blank'>here</a>",
// "ot_linux,ot_TensorFlow,ot_X64,ot_CUDA10":