"...ssh:/git@developer.sourcefind.cn:2222/OpenDAS/dynamo.git" did not exist on "37bc84444e8abea7e3e765929a78b069cd538900"
Unverified Commit 40ea0fb5 authored by Ryan McCormick's avatar Ryan McCormick Committed by GitHub
Browse files

docs: Add pinned install for cuda-python in pip install path (#2553)

parent d319abf3
......@@ -201,7 +201,10 @@ It is recommended to use [NGC PyTorch Container](https://catalog.ngc.nvidia.com/
### Install prerequisites
```
# Optional step: Only required for Blackwell and Grace Hopper
pip3 install torch==2.7.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
uv pip install torch==2.7.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
# Required until the trtllm version is bumped to include this pinned dependency itself
uv pip install "cuda-python>=12,<13"
sudo apt-get -y install libopenmpi-dev
```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment