Sections
Installation
I wanted to run ROCm on WSL so I could run some computer vision model training with Tensorflow which led me down a multi day rabbit hole.
- Install WSL (I'm partial to Ubuntu Server)
- Add these repos (might need to be changed depending on release)
sudo add-apt-repository -y -s deb http://security.ubuntu.com/ubuntu jammy main universe-
Install ROCm drivers for WSL here
-
Install tensorflow-rocm from here
-
Either export the following or add them to your .bashrc:
HSA_ENABLE_SDMA = 0;
ROCM_PATH = /opt/cmor;And it works!! Hopefully....