jan-swiatek
Nx / Axon - CUDA support
Hello,
I’ve just started learning Elixir few weeks ago and I really enjoy that language.
Currently, I’d like to start with machine learning / deep learning using Nx and Axon. I have an old NVIDIA GPU (NVS 510). In Python I could use PyTorch with CUDA, but for Elixir there aren’t that much tutorials showing how to set everything up. I’ve seen some articles about Axon and in all of them they use EXLA.
How to use full potential of my computer? Should I do everything just as it is shown in articles or is there a better way?
Most Liked
NduatiK
Have a look at this post https://forum.elixirforum.net/t/compiling-xla-with-cuda-11-2-support-needed/42528/4.
I usually have to set these env variables XLA_TARGET=cuda111 EXLA_TARGET=cuda LIBTORCH_TARGET=cuda
At the moment looks like Cuda 11.1 is prebuilt which might not be supported. From the post linekd above, you might have to do a custom Bazel build, but I’m not sure
Popular in Questions
Other popular topics
Latest on Elixir Forum
Sponsor Spotlight
The deployment platform built for Elixir: PaaS ease, VPS control.
Categories:
Sub Categories:
Forums
Our Sponsors
Build Elixir applications with speed and confidence.
Supporting innovation across the BEAM ecosystem.
We build reliable cloud platforms for business-critical systems.
Catch errors, track performance, monitor hosts and more.
Producing high quality Elixir screencasts since 2017.
Enabling companies to succeed by building software people love.
The deployment platform built for Elixir: PaaS ease, VPS control.
Courses that'll move you from confusion to "Aha, now I get it!"