Skip to content
System Status: The cluster is currently accessible but with reduced capacity available due to an emergency security maintenance
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/.