PyTorch

This is a quick guide to run PyTorch with ROCm support inside a provided docker image. See ROCm install for supported operating systems and general information on the ROCm software stack. A ROCm install version 2.1 is required currently. A Vega10 / gfx900 generation discrete graphics card is required (Vega56, Vega64, or MI25). The image contains hipified PyTorch source, a clone of the PyTorch examples, and has PyTorch for gfx900 installed.