"vscode:/vscode.git/clone" did not exist on "1d67149511404350c0c0bced73773afb6fcbfa21"
Unverified Commit 630479c3 authored by zhyncs's avatar zhyncs Committed by GitHub
Browse files

feat: update check env (#661)

parent 51fda143
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 ...@@ -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(): ...@@ -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}")
......
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