diff --git a/index.html b/index.html index 6e1971b4e9..3743a724bf 100644 --- a/index.html +++ b/index.html @@ -305,14 +305,11 @@

Platform

-

Platform list contains three items

+

Platform list contains one item

Linux
-
Windows
-
Mac
-
@@ -320,43 +317,25 @@

API

-

API list contains three items

+

API list contains one item

-
PyTorch 1.8.1
-
PyTorch 1.9
-
C++
-
-
-
- -
-
-

Architecture

-

Architecture list contains one item

-
-
-
-
- X64
+
ORTModule for Pytorch
+

Hardware Acceleration

-

Hardware Acceleration list contains four items

+

Hardware Acceleration list contains two items

-
Default CPU
-
CUDA 10.2
-
CUDA 11.1
-
ROCm 4.2 (Preview)
-
ROCm 4.3.1 (Preview)
-
oneDNN
+
CUDA
+
ROCm
diff --git a/js/script.js b/js/script.js index de71902adc..3d6b940089 100644 --- a/js/script.js +++ b/js/script.js @@ -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
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
python -m torch_ort.configure", + "ot_linux,ot_ORTModule,ot_X64,ot_CUDA": + "pip install onnxruntime-training -f https://downloads.onnxruntime.ai/onnxruntime_stable_<cu_version*>.html
pip install torch-ort
python -m torch_ort.configure

*Available versions", - "ot_linux,ot_PyTorch18,ot_X64,ot_DefaultCPU": - "Follow sample notebook from here", - - "ot_linux,ot_PyTorch18,ot_X64,ot_ROCm42": - "This combination requires an AMD docker image. See the torch-ort repo for details", - - "ot_linux,ot_PyTorch18,ot_X64,ot_ROCm431": - "This combination requires an AMD docker image. See the torch-ort repo for details", - - "ot_linux,ot_PyTorch18,ot_X64,ot_DNNL": - "This combination of resources is not fully tested. It may be possible to build from source.", - - "ot_linux,ot_PyTorch19,ot_X64,ot_CUDA10": - "pip install torch-ort
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
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
pip install torch-ort -f https://download.onnxruntime.ai/onnxruntime_stable_rocm42.html
python -m torch_ort.configure
", - - "ot_linux,ot_PyTorch19,ot_X64,ot_ROCm431": - "This combination requires an AMD docker image. See the torch-ort repo for details", - - "ot_linux,ot_C++,ot_X64,ot_CUDA10": - "This combination of resources is not fully tested. It may be possible to build from source.", - - "ot_linux,ot_C++,ot_X64,ot_DefaultCPU": - "This combination of resources is not fully tested. It may be possible to build from source.", + "ot_linux,ot_ORTModule,ot_X64,ot_ROCm": + "pip install onnxruntime-training -f https://downloads.onnxruntime.ai/onnxruntime_stable_<rocm_version*>.html
pip install torch-ort
python -m torch_ort.configure

*Available versions" - //windows - "ot_windows,ot_PyTorch18,ot_X64,ot_CUDA10": - "This combination of resources is not fully tested. It may be possible to build from source.", - - "ot_windows,ot_PyTorch18,ot_X64,ot_DefaultCPU": - "This combination of resources is not fully tested. It may be possible to build from source.", - - "ot_windows,ot_C++,ot_X64,ot_DefaultCPU": - "This combination of resources is not fully tested. It may be possible to build from source.", - - "ot_windows,ot_C++,ot_X64,ot_CUDA10": - "This combination of resources is not fully tested. It may be possible to build from source.", - - //mac - "ot_mac,ot_PyTorch18,ot_X64,ot_DefaultCPU": - "This combination of resources is not fully tested. It may be possible to build from source.", - - "ot_mac,ot_C++,ot_X64,ot_DefaultCPU": - "This combination of resources is not fully tested. It may be possible to build from source." }; 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 here", // "ot_linux,ot_TensorFlow,ot_X64,ot_CUDA10":