From 4859156a4413d1f87d43c6f7fa8f12ecdfe49606 Mon Sep 17 00:00:00 2001 From: Sophie Schoenmeyer <107952697+sophies927@users.noreply.github.com> Date: Wed, 3 Apr 2024 14:10:53 -0700 Subject: [PATCH] Add CUDA 12 instructions to get started table (#20170) Get Started table currently only includes instructions for Windows/Python/X64/CUDA and Linux/Python/X64/CUDA for (default) CUDA 11.8. Updating to include installation instructions for CUDA 12.X as well. --- src/routes/getting-started/table.svelte | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/routes/getting-started/table.svelte b/src/routes/getting-started/table.svelte index 4554db0f4f9e0..7b96daada60c6 100644 --- a/src/routes/getting-started/table.svelte +++ b/src/routes/getting-started/table.svelte @@ -77,31 +77,31 @@ const TrainingBuildIds = ['ot_stable', 'ot_nightly']; const validCombos = { 'windows,C-API,X64,CUDA': - "Install Nuget package Microsoft.ML.OnnxRuntime.Gpu
Refer to docs for requirements.", + "For CUDA 11.X (default): Install Nuget package Microsoft.ML.OnnxRuntime.Gpu.
For CUDA 12.X: Follow instructions here.
Refer to docs for requirements.", 'windows,C++,X64,CUDA': - "Install Nuget package Microsoft.ML.OnnxRuntime.Gpu
Refer to docs for requirements.", + "For CUDA 11.X (default): Install Nuget package Microsoft.ML.OnnxRuntime.Gpu.
For CUDA 12.X: Follow instructions here.
Refer to docs for requirements.", 'windows,C#,X64,CUDA': - "Install Nuget package Microsoft.ML.OnnxRuntime.Gpu
Refer to docs for requirements.", + "For CUDA 11.X (default): Install Nuget package Microsoft.ML.OnnxRuntime.Gpu.
For CUDA 12.X: Follow instructions here.
Refer to docs for requirements.", 'windows,Python,X64,CUDA': - "pip install onnxruntime-gpu
Refer to docs for requirements.", + "For CUDA 11.X (default):
pip install onnxruntime-gpu
For CUDA 12.X:
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
Refer to docs for requirements.", 'linux,Python,ARM64,CUDA': - "For Jetpack 4.4+, follow installation instructions from here", + "For Jetpack 4.4+, follow installation instructions from here.
Note: We do not have CUDA 12.X ARM64 support at this time.", 'linux,C-API,X64,CUDA': "Download .tgz file from Github
Refer to docs for requirements.", 'linux,C++,X64,CUDA': - "Install Nuget package Microsoft.ML.OnnxRuntime.Gpu
Refer to docs for requirements.", + "For CUDA 11.X (default): Install Nuget package Microsoft.ML.OnnxRuntime.Gpu.
For CUDA 12.X: Follow instructions here.
Refer to docs for requirements.", 'linux,C#,X64,CUDA': - "Install Nuget package Microsoft.ML.OnnxRuntime.Gpu
Refer to docs for requirements.", + "For CUDA 11.X (default): Install Nuget package Microsoft.ML.OnnxRuntime.Gpu.
For CUDA 12.X: Follow instructions here.
Refer to docs for requirements.", 'linux,Python,X64,CUDA': - "pip install onnxruntime-gpu
Refer to docs for requirements.", + "For CUDA 11.X (default):
pip install onnxruntime-gpu
For CUDA 12.X:
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
Refer to docs for requirements.", 'linux,C-API,ARM32,DefaultCPU': "Follow build instructions from here", @@ -335,7 +335,7 @@ "Add a dependency on com.microsoft.onnxruntime:onnxruntime using Maven/Gradle", 'linux,Java,X64,CUDA': - "Add a dependency on com.microsoft.onnxruntime:onnxruntime_gpu using Maven/Gradle.
Refer to docs for requirements.", + "Add a dependency on com.microsoft.onnxruntime:onnxruntime_gpu using Maven/Gradle.
Refer to docs for requirements.
Note: CUDA 12.X is not currently available with the Java GPU package due to low demand.", 'mac,Java,X64,DefaultCPU': "Add a dependency on com.microsoft.onnxruntime:onnxruntime using Maven/Gradle", @@ -382,7 +382,7 @@ "Add a dependency on com.microsoft.onnxruntime:onnxruntime using Maven/Gradle", 'windows,Java,X64,CUDA': - "Add a dependency on com.microsoft.onnxruntime:onnxruntime_gpu using Maven/Gradle.
Refer to docs for requirements.", + "Add a dependency on com.microsoft.onnxruntime:onnxruntime_gpu using Maven/Gradle.
Refer to docs for requirements.
Note: CUDA 12.X is not currently available with the Java GPU package due to low demand.", 'windows,Java,X64,TensorRT': "Add a dependency on com.microsoft.onnxruntime:onnxruntime_gpu using Maven/Gradle.
Refer to docs for usage details.",