Description
Provides a PyTorch build with ROCm GPU support and AVX2 CPU optimizations. It helps compatible AMD GPU systems accelerate tensor computation, training, and inference from Python.
ROCm support depends on GPU model, kernel, driver stack, and memory availability. Confirm compatibility before planning large machine-learning workloads.