fix(utils): change logic of setup_env to provide better feeling (Megv…

…ii-BaseDetection#63)

yolox/utils/setup_env.py

@@ -3,7 +3,6 @@
 # Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 
 import os
-import resource
 import subprocess
 
 import cv2
@@ -28,13 +27,24 @@ def configure_module(ulimit_value=8192):
     Configure pytorch module environment. setting of ulimit and cv2 will be set.
 
     Args:
-        ulimit_value(int): default open file number on linux. Default value: 4096.
+        ulimit_value(int): default open file number on linux. Default value: 8192.
     """
     # system setting
-    rlimit = resource.getrlimit(resource.RLIMIT_NOFILE)
-    resource.setrlimit(resource.RLIMIT_NOFILE, (ulimit_value, rlimit[1]))
+    try:
+        import resource
+        rlimit = resource.getrlimit(resource.RLIMIT_NOFILE)
+        resource.setrlimit(resource.RLIMIT_NOFILE, (ulimit_value, rlimit[1]))
+    except Exception:
+        # Exception might be raised in Windows OS or rlimit reaches max limit number.
+        # However, set rlimit value might not be necessary.
+        pass
+
     # cv2
     # multiprocess might be harmful on performance of torch dataloader
     os.environ["OPENCV_OPENCL_RUNTIME"] = "disabled"
-    cv2.setNumThreads(0)
-    cv2.ocl.setUseOpenCL(False)
+    try:
+        cv2.setNumThreads(0)
+        cv2.ocl.setUseOpenCL(False)
+    except Exception:
+        # cv2 version mismatch might rasie exceptions.
+        pass