Publish linux/arm64

[codingjoe]

Hi there 👋,

The upstream Nvidia image is available for arm64 too.
Let's give those arm users a hand… 😉

Best
Joe

PS: It would be cool if someone enabled the "package" tab on the GitHub repo so people could easily find the container image.

[devnen]

Thanks for the PR. I appreciate you thinking about arm64 support. The thing is, while the NVIDIA CUDA base image does have arm64 builds, I'm not confident our Python GPU dependencies would work reliably on that platform. Adding it would also mean setting up QEMU in CI, and honestly I don't have the hardware to properly test arm64 + NVIDIA setups right now. I'd rather keep things stable on amd64 where I know everything works than risk introducing issues I can't debug. If you need to run this on ARM hardware, you could try building locally with RUNTIME=cpu and see how it goes. Hope that makes sense