Directml Python Packages

torch-amd-setup

Auto-detect AMD GPU for PyTorch — ROCm, DirectML, CUDA, MPS, CPU. Fixes gfx1010 (RX 5700 XT) HSA_OVERRIDE_GFX_VERSION automatically. Windows/Linux/macOS/WSL2.

gpu-doctor

Universal GPU setup for PyTorch and JAX — DirectML, ROCm, CUDA, MPS, CPU. Auto-detects hardware, fixes HSA_OVERRIDE_GFX_VERSION for AMD RX 5700 XT. Beats devicetorch + torchruntime combined.

polyinfer

Unified ML inference across multiple backends: ONNX Runtime, OpenVINO, TensorRT, IREE