Machine Learning with AMD GPUs

Photo by Lubo Minar on Unsplash

You want to run PyTorch or TensorFlow on your PC, but you have an AMD GPU. No worries, this guide will get you up and running in no time!
Use Ubuntu LTS to get started as soon as possible.

First, install ROCm, the AMD equivalent of CUDA.

Then, install your favorite machine learning library.
The easiest way is to use docker.

For Pytorch, follow this:

For Tensorflow, follow this:

Now to start development, I recommend using Visual Studio Code (VSC).

Method 1:

First, start Jupyter in your container, copy the URI.


Now, configure your VSC to use the remote Jupyter server: for more details, follow this link:

Method 2:

Or, install the “Remote — Containers” extension for VSC.

ext install ms-vscode-remote.remote-containers

Happy hacking!