gpu-container-setup-flagos
SKILL.md
GPU Container Setup Skill
This skill automates multi-vendor GPU container setup for PyTorch workloads.
Supported GPU Vendors
| Vendor | PyTorch Backend | Detection |
|---|---|---|
| NVIDIA | CUDA | nvidia-smi |
| AMD | ROCm (HIP) | rocm-smi, /opt/rocm |
| Ascend | torch_npu | npu-smi, /usr/local/Ascend |
| Metax | torch_musa | mx-smi, /opt/metax |
| Iluvatar | torch_corex | ixsmi, /opt/iluvatar |
Execution Flow
When invoked, follow these steps: