Unverified Commit 630479c3 authored by zhyncs's avatar zhyncs Committed by GitHub
Browse files

feat: update check env (#661)

parent 51fda143
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}")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment