GPU support (plans/roadmap)

[asmith26]

Hi there,

I'm thinking about buying a new GPU soon for neural network training/fine-tuning, and was wondering if mojo has plans to support (or currently supports) the various options:

• NVIDIA (including e.g. any particular architectures/generations, CUDA versions/CUDA capability)
• AMD (including e.g. any particular architectures, ROCm versions)
• Intel (including e.g. any particular architectures/generations, oneAPI)

Many thanks for any help! :)

[gryznar]

As far as I know, support for Nvidia GPU is coming with next Max release

[ematejska]

Yes, exactly! Check this out for more info: https://docs.modular.com/max/faq#what-hardware-is-currently-supported-by-max-engine.

[asmith26]

Thanks that's really helpful. Are there plans to support AMD and Intel GPUs at some point? Thanks again