[bugfix] fix blackwell deepep installation (#22255)

This commit is contained in:
youkaichao
2025-08-06 01:26:09 +08:00
committed by GitHub
parent 469b3ffaaa
commit 59a0b8554b
2 changed files with 12 additions and 6 deletions

View File

@@ -13,16 +13,16 @@ All scripts accept a positional argument as workspace path for staging the build
## Usage
### Single-node
```bash
bash install_python_libraries.sh
# for hopper
TORCH_CUDA_ARCH_LIST="9.0" bash install_python_libraries.sh
# for blackwell
TORCH_CUDA_ARCH_LIST="10.0" bash install_python_libraries.sh
```
### Multi-node
Additional step for multi-node deployment:
```bash
bash install_python_libraries.sh
sudo bash configure_system_drivers.sh
sudo reboot # Reboot is required to load the new driver
```