Devices and outputs¶
Device selection is centralized in utils/device_utils.py.
Detection¶
detect_compute_environment() checks:
- whether
nvidia-smiexists - whether the system is macOS
- whether the machine is Apple Silicon
- whether PyTorch MPS is built and available
Preference:
This mainly affects PyTorch models. XGBoost, LightGBM, and CatBoost only enable GPU parameters on CUDA branches.
Output files¶
AutoML:
AutoDL:
save_results() creates the output directory if needed and writes UTF-8 JSON with indentation.