Install ONNX Runtime GenAI
Python package release candidates
pip install numpy
pip install onnxruntime-genai --pre --index-url=https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-genai/pypi/simple/
Append -cuda
for the library that is optimized for CUDA environments
pip install onnxruntime-genai-cuda --pre --index-url=https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-genai/pypi/simple/
Nuget package release candidates
To install the NuGet release candidates, add a new package source in Visual Studio, go to Project
-> Manage NuGet Packages
.
-
Click on the
Settings
cog icon -
Click the
+
button to add a new package source- Change the Name to
onnxruntime-genai
- Change the Source to
https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-genai/nuget/v3/index.json
- Change the Name to
-
Check the
Include prerelease
button -
Add the
Microsoft.ML.OnnxRuntimeGenAI
package -
Add the
Microsoft.ML.OnnxRuntime
package
To run with CUDA, use the following packages instead:
Microsoft.ML.OnnxRuntimeGenAI.Cuda
Microsoft.ML.OnnxRuntime.Gpu