diff --git a/docs/install/index.md b/docs/install/index.md index 9ad2e13531c8a..8d84ba551c701 100644 --- a/docs/install/index.md +++ b/docs/install/index.md @@ -70,25 +70,32 @@ pip install flatbuffers numpy packaging protobuf sympy pip install --pre --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-directml ``` -### Install ONNX Runtime GPU (CUDA 12.x) +### Install ONNX Runtime GPU (CUDA or TensorRT) +#### CUDA 12.x The default CUDA version for [onnxruntime-gpu in pypi](https://pypi.org/project/onnxruntime-gpu) is 12.x since 1.19.0. ```bash pip install onnxruntime-gpu ``` -#### Install nightly +For previous versions, you can download here: [1.18.1](https://aiinfra.visualstudio.com/PublicPackages/_artifacts/feed/onnxruntime-cuda-12/PyPI/onnxruntime-gpu/overview/1.18.1), [1.18.0](https://aiinfra.visualstudio.com/PublicPackages/_artifacts/feed/onnxruntime-cuda-12/PyPI/onnxruntime-gpu/overview/1.18.0) + +#### Nightly for CUDA 13.x ```bash pip install flatbuffers numpy packaging protobuf sympy -pip install --pre --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-gpu +pip install --pre --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ort-cuda-13-nightly/pypi/simple/ onnxruntime-gpu ``` -For previous versions, you can download here: [1.18.1](https://aiinfra.visualstudio.com/PublicPackages/_artifacts/feed/onnxruntime-cuda-12/PyPI/onnxruntime-gpu/overview/1.18.1), [1.18.0](https://aiinfra.visualstudio.com/PublicPackages/_artifacts/feed/onnxruntime-cuda-12/PyPI/onnxruntime-gpu/overview/1.18.0) +#### Nightly for CUDA 12.x +```bash +pip install flatbuffers numpy packaging protobuf sympy +pip install --pre --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ onnxruntime-gpu +``` -### Install ONNX Runtime GPU (CUDA 11.x) +#### CUDA 11.x For Cuda 11.x, please use the following instructions to install from [ORT Azure Devops Feed](https://aiinfra.visualstudio.com/PublicPackages/_artifacts/feed/onnxruntime-cuda-11/PyPI/onnxruntime-gpu/overview) for 1.19.2 or later. @@ -99,7 +106,6 @@ pip install onnxruntime-gpu --index-url https://aiinfra.pkgs.visualstudio.com/Pu For previous versions, you can download here: [1.18.1](https://pypi.org/project/onnxruntime-gpu/1.18.1/), [1.18.0](https://pypi.org/project/onnxruntime-gpu/1.18.0/) - ### Install ONNX Runtime QNN ```bash