Skip to content

Commit

Permalink
add option for python training nightly installation for cuda 12 (#19246)
Browse files Browse the repository at this point in the history
Add an option for installing ort-training nightly build for cuda 12.2
scenario

### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
  • Loading branch information
ajindal1 authored Jan 24, 2024
1 parent e679d99 commit 73356fc
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions src/routes/getting-started/table.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -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 = {
Expand Down Expand Up @@ -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<br/>pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-training<br/>pip install torch-ort<br/>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<br/>pip install -i https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-12-nightly/pypi/simple/ onnxruntime-training<br/>pip install torch-ort<br/>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_<b>&lt;rocm_version*</b>&gt;.html<br/>pip install torch-ort<br/>python -m torch_ort.configure<br/><br/>*<a class='text-blue-500' href='https://download.onnxruntime.ai/' target='blank'>Available versions</a>",
Expand Down Expand Up @@ -2018,7 +2021,7 @@
aria-labelledby="ot_selectHardwareAcceleration"
aria-describedby="ot_decriptionHardwareAcceleration"
>
<div class="grid grid-cols-3 gap-4 ot_hardwareAcceleration">
<div class="grid grid-cols-4 gap-4 ot_hardwareAcceleration">
{#each TrainingVersions as version, i}
<a
class="r-option version join-item btn rounded"
Expand Down

0 comments on commit 73356fc

Please sign in to comment.