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

adding cuda 12 ort training for stable and on-device scenarios (#19439)

Browse source 
### Description
<!-- Describe your changes. -->
adding cuda 12 ort training for stable and on-device scenarios. Also,
there was a removal for CUDA 12 in this
[PR](https://github.com/microsoft/onnxruntime/pull/19342), fixing it.


### 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. -->
This commit is contained in:
Abhishek Jindal 2024-02-06 10:03:30 -08:00 committed by GitHub
parent fde74cbb00
commit bd63813dc1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/routes/getting-started/table.svelte
View file

@ -56,8 +56,8 @@
const TrainingPlatformIds = ['ot_linux', 'ot_windows', 'ot_mac', 'ot_android', 'ot_ios', 'ot_web'];
const TrainingAPIs = ['Python', 'C', 'C++', 'C#', 'Java', 'Obj-C', 'JavaScript'];
const TrainingAPIIds = ['ot_python', 'ot_c', 'ot_cplusplus', 'ot_csharp', 'ot_java', 'ot_objc', 'ot_js'];
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 = {
@ -706,6 +706,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_stable':
'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/onnxruntime-cuda-12/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',
@ -727,6 +730,12 @@
'ot_linux,ot_on_device,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',
'ot_linux,ot_on_device,ot_python,ot_X64,ot_CUDA122,ot_stable':
'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/onnxruntime-cuda-12/pypi/simple/ onnxruntime-training',
'ot_linux,ot_on_device,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',
'ot_linux,ot_on_device,ot_cplusplus,ot_X64,ot_CPU,ot_stable':
"Download .tgz file from&nbsp;<a class='text-blue-500' href='https://github.com/microsoft/onnxruntime/releases' target='_blank'>Github</a> <br/>Refer to <a class='text-blue-500' href='http://www.onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements' target='_blank'>docs</a> for requirements.",