Skip to content

Setup to use MIGraphX with docker

Paul Fultz II edited this page Jul 7, 2021 · 2 revisions

To install rocm, I suggest to have ubuntu 18.04 and linux 5.4(although 20.04 should work as long as you can install 5.4 linux kernel). You can check linux version you have installed with uname -a.

Before building the docker you will need to install at least the driver on the host machine. This can be done with:

sh -c 'echo deb [arch=amd64 trusted=yes] http://repo.radeon.com/rocm/apt/4.2/ xenial main > /etc/apt/sources.list.d/rocm.list'
sudo apt-get update
sudo apt-get -y install rocm-dkms && sudo reboot

Here is the migraphx.docker file.