Shortcuts

mmcv.utils.collect_env

mmcv.utils.collect_env()[源代码]

Collect the information of the running environments.

返回

The environment information. The following fields are contained.

  • sys.platform: The variable of sys.platform.

  • Python: Python version.

  • CUDA available: Bool, indicating if CUDA is available.

  • GPU devices: Device type of each GPU.

  • CUDA_HOME (optional): The env var CUDA_HOME.

  • NVCC (optional): NVCC version.

  • GCC: GCC version, “n/a” if GCC is not installed.

  • MSVC: Microsoft Virtual C++ Compiler version, Windows only.

  • PyTorch: PyTorch version.

  • PyTorch compiling details: The output of torch.__config__.show().

  • TorchVision (optional): TorchVision version.

  • OpenCV: OpenCV version.

  • MMEngine: MMEngine version.

  • MMCV: MMCV version.

  • MMCV Compiler: The GCC version for compiling MMCV ops.

  • MMCV CUDA Compiler: The CUDA version for compiling MMCV ops.

返回类型

dict