Feat: python -m zeus.show_env

[jaywonchung]

This adds an in-tree script show_env.py.

Users can run

$ python -m zeus.show_env

after installing Zeus with pip to understand whether Python packages and hardware (NVIDIA GPU, AMD GPU, Intel RAPL-supported CPU) are recognized.

[jaywonchung]

@wbjin Thanks! Handled cases where things are missing. The following is on my laptop:

$ python -m zeus.show_env
================================================================================

Python version: 3.9.19

================================================================================

[2024-09-09 17:35:04,811] [zeus.utils.framework](framework.py:24) PyTorch with CUDA support is available.
[2024-09-09 17:35:04,811] [zeus.utils.framework](framework.py:45) JAX is not available

Package availability and versions:
  Zeus: 0.10.0
  PyTorch: 2.4.0 (without CUDA support)
  JAX: not available

================================================================================

[2024-09-09 17:35:04,811] [zeus.device.gpu.nvidia](nvidia.py:49) pynvml is available but could not initialize NVML: NVML Shared Library Not Found.
[2024-09-09 17:35:04,811] [zeus.device.gpu.amd](amd.py:42) amdsmi is not available.

GPU availability:
  No GPUs available.

================================================================================

[2024-09-09 17:35:04,811] [zeus.device.cpu.rapl](rapl.py:134) RAPL is not supported on this CPU.

CPU availability:
  No CPUs available.

================================================================================

[wbjin]

Works on my laptop and cloudlab as well!