kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: convert venv_prefix to str before comparison with base_prefix (#6121)

Browse Source 
Signed-off-by: Zhenhuan Chen <chenzhh3671@gmail.com>
This commit is contained in:
Zhenhuan Chen 2025-07-17 15:04:54 +08:00 committed by GitHub
parent e821c68611
commit 8c1c9ef7aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/build_wheel.py
View File

@ -144,7 +144,7 @@ def setup_venv(project_dir: Path, requirements_file: Path, no_venv: bool):
# Ensure PyPI PyTorch is not installed in the venv
purelib_dir = Path(scheme["purelib"])
pytorch_package_dir = purelib_dir / "torch"
if venv_prefix != sys.base_prefix and pytorch_package_dir.exists():
if str(venv_prefix) != sys.base_prefix and pytorch_package_dir.exists():
warnings.warn(
f"Using the NVIDIA PyTorch container with PyPI distributed PyTorch may lead to compatibility issues.\n"
f"If you encounter any problems, please delete the environment at `{venv_prefix}` so that "