feat: update check env (#661)

This commit is contained in:
zhyncs
2024-07-20 02:54:15 +10:00
committed by GitHub
parent 51fda1439f
commit 630479c3a6

View File

@@ -1,5 +1,6 @@
import importlib
import os
import resource
import subprocess
import sys
from collections import OrderedDict, defaultdict
@@ -10,6 +11,9 @@ import torch
PACKAGE_LIST = [
"sglang",
"flashinfer",
"requests",
"tqdm",
"numpy",
"aiohttp",
"fastapi",
"hf_transfer",
@@ -154,6 +158,9 @@ def check_env():
if 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():
print(f"{k}: {v}")