fix: reinstall torch in deps install (#11414)
This commit is contained in:
2
Makefile
2
Makefile
@@ -24,7 +24,7 @@ FILES_TO_UPDATE = docker/Dockerfile.rocm \
|
|||||||
docs/get_started/install.md \
|
docs/get_started/install.md \
|
||||||
docs/platforms/amd_gpu.md \
|
docs/platforms/amd_gpu.md \
|
||||||
docs/platforms/ascend_npu.md \
|
docs/platforms/ascend_npu.md \
|
||||||
benchmark/deepseek_v3/README.md
|
benchmark/deepseek_v3/README.md
|
||||||
|
|
||||||
update: ## Update version numbers across project files. Usage: make update <new_version>
|
update: ## Update version numbers across project files. Usage: make update <new_version>
|
||||||
@if [ -z "$(filter-out $@,$(MAKECMDGOALS))" ]; then \
|
@if [ -z "$(filter-out $@,$(MAKECMDGOALS))" ]; then \
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ else
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# Install the main package
|
# Install the main package
|
||||||
$PIP_CMD install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} $PIP_INSTALL_SUFFIX
|
$PIP_CMD install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} $PIP_INSTALL_SUFFIX --force-reinstall
|
||||||
|
|
||||||
# Install router for pd-disagg test
|
# Install router for pd-disagg test
|
||||||
SGLANG_ROUTER_BUILD_NO_RUST=1 $PIP_CMD install -e "sgl-router" $PIP_INSTALL_SUFFIX
|
SGLANG_ROUTER_BUILD_NO_RUST=1 $PIP_CMD install -e "sgl-router" $PIP_INSTALL_SUFFIX
|
||||||
@@ -68,7 +68,7 @@ if [ "$IS_BLACKWELL" != "1" ]; then
|
|||||||
$PIP_CMD install -e lmms-eval/ $PIP_INSTALL_SUFFIX
|
$PIP_CMD install -e lmms-eval/ $PIP_INSTALL_SUFFIX
|
||||||
|
|
||||||
# Install xformers
|
# Install xformers
|
||||||
$PIP_CMD install xformers --index-url https://download.pytorch.org/whl/${CU_VERSION} --no-deps $PIP_INSTALL_SUFFIX
|
$PIP_CMD install xformers --index-url https://download.pytorch.org/whl/${CU_VERSION} --no-deps $PIP_INSTALL_SUFFIX --force-reinstall
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Show current packages
|
# Show current packages
|
||||||
|
|||||||
Reference in New Issue
Block a user