Train neural networks using AMD GPU and Keras

Getting started with ROCm platform

Mattia Varile
Towards Data Science
10 min readFeb 11, 2019

--

AMD is developing a new HPC platform, called ROCm. Its ambition is to create a common, open-source environment, capable to interface both with Nvidia (using CUDA) and AMD GPUs (further information).

This tutorial will explain how to set-up a neural network environment, using AMD GPUs in a single or…

--

--