View on GitHub ROCm_Logo

ROCm, a New Era in Open GPU Computing

Platform for GPU-Enabled HPC and Ultrascale Computing

Deep Learning on ROCm

ROCm Tensorflow v1.10 Release

We are excited to announce the release of ROCm enabled TensorFlow v1.11 for AMD GPUs.

Tensorflow Installation

First, you’ll need to install the open-source ROCm stack, require at least ROCm1.9.0. Details can be found here: Then, install these other relevant ROCm packages:

sudo apt update
sudo apt install rocm-libs miopen-hip cxlactivitylogger

And finally, install TensorFlow itself (via the Python Package Index):

sudo apt install wget python3-pip
# Pip3 install the whl package from PyPI
pip3 install --user tensorflow-rocm

Now that Tensorflow v1.11 is installed!

Tensorflow More Resources

Tensorflow docker images are also publicly available, more details can be found here:

Please connect with us for any questions, our official github repository is here:

ROCm MIOpen v1.5 Release

Announcing our new Foundation for Deep Learning acceleration MIOpen 1.5 which introduces support for Convolution Neural Network (CNN) acceleration — built to run on top of the ROCm software stack!

This release includes the following:

Porting from cuDNN to MIOpen

The porting guide highlights the key differences between the current cuDNN and MIOpen APIs.

The ROCm 1.9 has prebuilt packages for MIOpen

Install the ROCm MIOpen implementation (assuming you already have the ‘rocm’ and ‘rocm-opencl-dev” package installed):

For just OpenCL development

sudo apt-get install miopengemm miopen-opencl

For HIP development

sudo apt-get install miopengemm miopen-hip

Or you can build from source code

Deep Learning Framework support for ROCm

Framework Status MIOpen Enabled Upstreamed   Current Repository
Caffe       Public     Yes                
Tensorflow   Development Yes             CLA in Progress Notes: Working on NCCL and XLA enablement, Running
Caffe2       Upstreaming Yes CLA in Progress
Torch HIP   Upstreaming Development In process
HIPnn        Upstreaming Development          
PyTorch      Development Development    
MxNet       Development Development         
CNTK       Development Development