Update README.md

RTX 50xx support
pull/1361/head
Kenneth Estanislao 2025-06-15 02:22:00 +08:00 committed by GitHub
parent 745d449ca6
commit f121083bc8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -196,7 +196,7 @@ pip install -r requirements.txt
```bash ```bash
pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128 pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
pip uninstall onnxruntime onnxruntime-gpu pip uninstall onnxruntime onnxruntime-gpu
pip install onnxruntime-gpu==1.16.3 pip install onnxruntime-gpu==1.21.0
``` ```
3. Usage: 3. Usage: