From 630479c3a6acc8a36554843be2101e1e6ff04e8c Mon Sep 17 00:00:00 2001
From: zhyncs <me@zhyncs.com>
Date: Sat, 20 Jul 2024 02:54:15 +1000
Subject: [PATCH] feat: update check env (#661)

---
 python/sglang/check_env.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/python/sglang/check_env.py b/python/sglang/check_env.py
index f6a5d7aae..957fb2852 100644
--- a/python/sglang/check_env.py
+++ b/python/sglang/check_env.py
@@ -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}")