mirror of
https://github.com/JezzWTF/vibepod.git
synced 2026-06-01 15:22:14 +00:00
Fix ROCm torch wheel not replacing CPU torch
uv pip install without --reinstall-package silently skips the ROCm wheel when CPU torch already satisfies torch>=2.0.0, leaving a CPU installation in .venv-rocm and causing a broken import at startup. https://claude.ai/code/session_0168pSswiaoEf6LEx6UQWfBu
This commit is contained in:
+1
-1
@@ -105,7 +105,7 @@ elif $ROCM_MODE; then
|
||||
fi
|
||||
uv sync --no-sources
|
||||
echo "--> Installing PyTorch ROCm 6.2 wheel..."
|
||||
uv pip install torch --index-url https://download.pytorch.org/whl/rocm6.2
|
||||
uv pip install torch --index-url https://download.pytorch.org/whl/rocm6.2 --reinstall-package torch
|
||||
if [[ -n "$LOCK_BACKUP" ]]; then
|
||||
cp "$LOCK_BACKUP" uv.lock
|
||||
rm -f "$LOCK_BACKUP"
|
||||
|
||||
Reference in New Issue
Block a user