EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

feat: update check env (#661)

Browse Source
This commit is contained in:
zhyncs
2024-07-20 02:54:15 +10:00
committed by GitHub
parent 51fda1439f
commit 630479c3a6
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/sglang/check_env.py
View File

@@ -1,5 +1,6 @@
import importlib import importlib
import os import os
import resource
import subprocess import subprocess
import sys import sys
from collections import OrderedDict, defaultdict from collections import OrderedDict, defaultdict
@@ -10,6 +11,9 @@ import torch
PACKAGE_LIST = [ PACKAGE_LIST = [
"sglang", "sglang",
"flashinfer", "flashinfer",
"requests",
"tqdm",
"numpy",
"aiohttp", "aiohttp",
"fastapi", "fastapi",
"hf_transfer", "hf_transfer",
@@ -154,6 +158,9 @@ def check_env():
if gpu_topo: if gpu_topo:
env_info["NVIDIA Topology"] = gpu_topo env_info["NVIDIA Topology"] = gpu_topo
ulimit_soft, _ = resource.getrlimit(resource.RLIMIT_NOFILE)
env_info["ulimit soft"] = ulimit_soft
for k, v in env_info.items(): for k, v in env_info.items():
print(f"{k}: {v}") print(f"{k}: {v}")