From 0318e0ecb09daca31fc395287baf78ae9c7b5878 Mon Sep 17 00:00:00 2001 From: Abhishek Jindal Date: Tue, 23 Jan 2024 13:23:20 -0800 Subject: [PATCH 1/2] add option for python training nightly installation for large model with cuda 12.2 --- src/routes/getting-started/table.svelte | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/routes/getting-started/table.svelte b/src/routes/getting-started/table.svelte index 62ba55864589c..0e896348194b2 100644 --- a/src/routes/getting-started/table.svelte +++ b/src/routes/getting-started/table.svelte @@ -56,8 +56,8 @@ const TrainingPlatformIds = ['ot_linux', 'ot_windows', 'ot_mac', 'ot_android', 'ot_ios']; const TrainingAPIs = ['Python', 'C', 'C++', 'C#', 'Java', 'Obj-C']; const TrainingAPIIds = ['ot_python', 'ot_c', 'ot_cplusplus', 'ot_csharp', 'ot_java', 'ot_objc']; - const TrainingVersions = ['CUDA 11.8', 'ROCm', 'CPU']; - const TrainingVersionIds = ['ot_CUDA118', 'ot_ROCm', 'ot_CPU']; + const TrainingVersions = ['CUDA 11.8', 'CUDA 12.2', 'ROCm', 'CPU']; + const TrainingVersionIds = ['ot_CUDA118', 'ot_CUDA122', 'ot_ROCm', 'ot_CPU']; const TrainingBuilds = ['Stable', 'Preview (Nightly)']; const TrainingBuildIds = ['ot_stable', 'ot_nightly']; const validCombos = { @@ -690,6 +690,9 @@ 'ot_linux,ot_large_model,ot_python,ot_X64,ot_CUDA118,ot_nightly': 'python -m pip install cerberus flatbuffers h5py numpy>=1.16.6 onnx packaging protobuf sympy setuptools>=41.4.0
pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-training
pip install torch-ort
python -m torch_ort.configure', + 'ot_linux,ot_large_model,ot_python,ot_X64,ot_CUDA122,ot_nightly': + 'python -m pip install cerberus flatbuffers h5py numpy>=1.16.6 onnx packaging protobuf sympy setuptools>=41.4.0
pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/ onnxruntime-training
pip install torch-ort
python -m torch_ort.configure', + 'ot_linux,ot_large_model,ot_python,ot_X64,ot_ROCm,ot_stable': "pip install onnxruntime-training -f https://download.onnxruntime.ai/onnxruntime_stable_<rocm_version*>.html
pip install torch-ort
python -m torch_ort.configure

*Available versions", From 98e402527b5b9273fa438b42c9d8a4863e8f3ee9 Mon Sep 17 00:00:00 2001 From: Abhishek Jindal Date: Tue, 23 Jan 2024 20:53:43 -0800 Subject: [PATCH 2/2] update number of cols --- src/routes/getting-started/table.svelte | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/routes/getting-started/table.svelte b/src/routes/getting-started/table.svelte index 0e896348194b2..2e8f4cc178582 100644 --- a/src/routes/getting-started/table.svelte +++ b/src/routes/getting-started/table.svelte @@ -2021,7 +2021,7 @@ aria-labelledby="ot_selectHardwareAcceleration" aria-describedby="ot_decriptionHardwareAcceleration" > -
+
{#each TrainingVersions as version, i}