Enable running collect_env.py without building PyTorch (#15468)
Summary: Closes #15346
Differential Revision: D13537873
Pulled By: ezyang
fbshipit-source-id: 7765ce4108dae9479d8900c0815cc2f174596a83
diff --git a/torch/utils/collect_env.py b/torch/utils/collect_env.py
index 0488995..b406629 100644
--- a/torch/utils/collect_env.py
+++ b/torch/utils/collect_env.py
@@ -9,7 +9,11 @@
import os
from collections import namedtuple
-import torch
+try:
+ import torch
+ TORCH_AVAILABLE = True
+except (ImportError, NameError, AttributeError):
+ TORCH_AVAILABLE = False
PY3 = sys.version_info >= (3, 0)
@@ -221,12 +225,20 @@
run_lambda = run
pip_version, pip_list_output = get_pip_packages(run_lambda)
+ if TORCH_AVAILABLE:
+ version_str = torch.__version__
+ debug_mode_str = torch.version.debug
+ cuda_available_str = torch.cuda.is_available()
+ cuda_version_str = torch.version.cuda
+ else:
+ version_str = debug_mode_str = cuda_available_str = cuda_version_str = 'N/A'
+
return SystemEnv(
- torch_version=torch.__version__,
- is_debug_build=torch.version.debug,
+ torch_version=version_str,
+ is_debug_build=debug_mode_str,
python_version='{}.{}'.format(sys.version_info[0], sys.version_info[1]),
- is_cuda_available=torch.cuda.is_available(),
- cuda_compiled_version=torch.version.cuda,
+ is_cuda_available=cuda_available_str,
+ cuda_compiled_version=cuda_version_str,
cuda_runtime_version=get_running_cuda_version(run_lambda),
nvidia_gpu_models=get_gpu_info(run_lambda),
nvidia_driver_version=get_nvidia_driver_version(run_lambda),
@@ -308,7 +320,7 @@
all_cuda_fields = dynamic_cuda_fields + ['cudnn_version']
all_dynamic_cuda_fields_missing = all(
mutable_dict[field] is None for field in dynamic_cuda_fields)
- if not torch.cuda.is_available() and all_dynamic_cuda_fields_missing:
+ if TORCH_AVAILABLE and not torch.cuda.is_available() and all_dynamic_cuda_fields_missing:
for field in all_cuda_fields:
mutable_dict[field] = 'No CUDA'
if envinfo.cuda_compiled_version is None:
