Skip to content
System Status: All systems are operational • Services are available and operational.
Click for detailed status

ROCm

AMD ROCm is an open software stack including drivers, development tools, and APIs that enable GPU programming from low-level kernel to end-user applications. ROCm is optimized for Generative AI and HPC applications, and is easy to migrate existing code into.

PyTorch

To install a ROCm compatible version of PyTorch, follow the instructions on https://pytorch.org/.

TensorFlow

To install a ROCm compatible version if TensorFlow, use https://pypi.org/project/tensorflow-rocm/.