Publish a Docker Container with ROCM

xuhuisheng / rocm-build

build scripts for ROCm

Apache License 2.0

181 stars 35 forks source link

Publish a Docker Container with ROCM #42

Open guilt opened 1 year ago

guilt commented 1 year ago

Hi,

First of all, I wanted to say - great work!

This is not a issue technically, but would you be able to build a Docker image with the fully patched toolchain for ROCM 5.4.x. That way a lot of folks can use it to test and run PyTorch/AI stuff on the older cards.

See: https://gist.github.com/guilt/6c901f7ac0a726685b6334798da77c00

Would be great to use your patched .deb files and install them so it could be tested on older GPUs.

xuhuisheng commented 1 year ago

I am a little busy now. I think I can take a try for hub.docker.com later.

serhii-nakon commented 10 months ago

@guilt Hello here Docker image for RX5500 https://hub.docker.com/r/serhiin/rocm_gfx1012_pytorch

guilt commented 10 months ago

Please publish this, share it on HN/ROCM Wikipedia page. You're doing everyone a favor. Do post screenshots when you can run it on the said hardware. Thank you 🙏

I'm keeping this issue as a documentation. In case you do build with gfx803 enabled, that would maximize the compatibiity. Please publish that image too.