How do I make my AMD GPU available within a docker image based on python3.9.10

HOW TO -️ October 18, 2021

I'd like to do some machine learning on my AMD 6800 XT gpu within a python image based on python:3.9.10. I can confirm that the GPU is available outside of the image (in a wsl2 instance). However, if I do docker run -it python:3.9.10 /bin/bash and then complete the same tutorial ( it doesn't work:

ERROR Rendering Code Block

This article has led me to think that perhaps docker doesn't support AMD GPUs at all:

Can anyone suggest what I might be able to do to get this to work?

Note that the reason I have picked this image is that my environment is based on a rather lengthy Dockerfile inheriting from python:3.9.10, and I'd like to keep using that image on the PC with the GPU as well as other (nvidia) environments, so I'm after a portable solution as far as the image is concerned, although I'd be grateful for any solution at this point.


Basically you are just passing it in with the --device flag but you need drivers and stuff. Check this project out the details section is really helpful.