xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-11 13:09:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix] python collect_env.py and vllm collect-env compatibility with uv venv (#24066)

Browse Source 
Signed-off-by: Kay Yan <kay.yan@daocloud.io>
This commit is contained in:
Kay Yan 2025-09-10 21:25:33 +08:00 committed by GitHub
parent ccee371e86
commit 2eb9986a2d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vllm/collect_env.py
View File

@ -489,6 +489,16 @@ def get_libc_version():
return '-'.join(platform.libc_ver()) return '-'.join(platform.libc_ver())
def is_uv_venv():
if os.environ.get("UV"):
return True
pyvenv_cfg_path = os.path.join(sys.prefix, 'pyvenv.cfg')
if os.path.exists(pyvenv_cfg_path):
with open(pyvenv_cfg_path, 'r') as f:
return any(line.startswith('uv = ') for line in f)
return False
def get_pip_packages(run_lambda, patterns=None): def get_pip_packages(run_lambda, patterns=None):
"""Return `pip list` output. Note: will also find conda-installed pytorch and numpy packages.""" """Return `pip list` output. Note: will also find conda-installed pytorch and numpy packages."""
if patterns is None: if patterns is None:
@ -504,7 +514,7 @@ def get_pip_packages(run_lambda, patterns=None):
if pip_available: if pip_available:
cmd = [sys.executable, '-mpip', 'list', '--format=freeze'] cmd = [sys.executable, '-mpip', 'list', '--format=freeze']
elif os.environ.get("UV") is not None: elif is_uv_venv():
print("uv is set") print("uv is set")
cmd = ["uv", "pip", "list", "--format=freeze"] cmd = ["uv", "pip", "list", "--format=freeze"]
else: else: