diff --git a/src/routes/getting-started/table.svelte b/src/routes/getting-started/table.svelte index 62ba558645..2e8f4cc178 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", @@ -2018,7 +2021,7 @@ aria-labelledby="ot_selectHardwareAcceleration" aria-describedby="ot_decriptionHardwareAcceleration" > -
+
{#each TrainingVersions as version, i}