added the model

roadcast/check_env.py

@@ -0,0 +1,32 @@
+import sys
+import importlib
+
+def safe_import(name):
+    try:
+        return importlib.import_module(name)
+    except Exception as e:
+        return e
+
+print('Python:', sys.version.replace('\n',' '))
+
+torch = safe_import('torch')
+if isinstance(torch, Exception):
+    print('torch import error:', torch)
+else:
+    print('torch:', torch.__version__)
+    print('CUDA available:', torch.cuda.is_available())
+    if torch.cuda.is_available():
+        print('CUDA device count:', torch.cuda.device_count())
+        print('Current device name:', torch.cuda.get_device_name(0))
+
+pandas = safe_import('pandas')
+if isinstance(pandas, Exception):
+    print('pandas import error:', pandas)
+else:
+    print('pandas:', pandas.__version__)
+
+try:
+    import sklearn
+    print('sklearn:', sklearn.__version__)
+except Exception:
+    pass