-
Notifications
You must be signed in to change notification settings - Fork 85
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.