4 ms·
I just want an alternative to CUDA. I am sick of Nvidia’s monopoly in this space.
by lvl100 5y ago
I just want an alternative to CUDA. I am sick of Nvidia’s monopoly in this space.
- nestorD 5y agoI believe all big frameworks have some work being done to make them compatible with AMD GPUs. Here is the relevant issue for JAX (support seem to be in alpha but viable): https://github.com/google/jax/issues/2012 https://github.com/google/jax/issues/2012
- YetAnotherNick 5y agoThe best thing that worked for me is the Apple's tensorflow pluggable device for metal. It could utilize both AMD and M1 GPU to 100% capacity. It's a shame apple could do it as a small project in the side to use metal but AMD couldn't do the same with Vulkan.