A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 15, 18:30 UTC
[ROCM] Upgrade torch to 2.6 (#15244)
Signed-off-by: Sage Moore <sage@neuralmagic.com>
Sage Moore committed
Mar 20, 2025 at 19:17 UTC
2e0b4cfde04585ef3acb9257d9f7b38776da926d
1 file changed
+4
-4
requirements/rocm-build.txt
+4
-4
@@ -1,10 +1,10 @@
1
# Common dependencies
2
-r common.txt
3
4
---extra-index-url https://download.pytorch.org/whl/rocm6.2
5
-torch==2.5.1
6
-torchvision==0.20.1
7
-torchaudio==2.5.1
4
+--extra-index-url https://download.pytorch.org/whl/rocm6.2.4
5
+torch==2.6.0
6
+torchvision==0.21.0
7
+torchaudio==2.6.0
8
9
cmake>=3.26
10
packaging