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.
-
Updated
Mar 29, 2026 - Python
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.
Reproducible GPU float32 benchmarks — AMD DirectML 40.2x speedup on RX 5700 XT. Windows/Linux/macOS. torch-directml, ROCm, CUDA, MPS. All results verified on real hardware.
基于torch-directml开发的常见模型的训练脚本
Meaningful Speech Enhancement (Noise and Reverb Suppression) | Native-Friendly to torch_directml and Fastai on Windows 10/11
🌐 Streamline your downloads with PytDm, a modern Python download manager featuring a user-friendly interface, pause/resume options, and real-time progress tracking.
Add a description, image, and links to the torch-directml topic page so that developers can more easily learn about it.
To associate your repository with the torch-directml topic, visit your repo's landing page and select "manage topics."