jan-swiatek
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?
Where Next?
Trending in Questions
Other Trending Topics
Latest on Elixir Forum
Sponsor Spotlight
Elixir, Erlang & messaging conferences from Code Sync. Next up: ElixirConf US (10–11 Sep) & Code BEAM Europe (21–22 Oct), Haarlem + online.
Categories:
Sub Categories:
Forums
Popular Tags
- #ecto
- #liveview
- #troubleshooting
- #learning-elixir
- #deployment
- #library
- #erlang
- #testing
- #genserver
- #mix
- #absinthe
- #remote-other
- #otp
- #plug
- #how-to-question
- #macros
- #postgres
- #elixirconf
- #channels
- #exunit
- #discussion
- #code-sync
- #podcasts
- #javascript
- #onsite
- #dialyzer
- #docker
- #authentication
- #umbrella
- #full-time-contract
- #podcasts-by-brainlid
- #ecto-query
- #blog-post
- #elixirconf-us
- #elixir-ls
- #ai
- #phoenix_html
- #iex
- #graphql
- #genstage
- #websockets
- #supervisor
- #advent-of-code
- #distillery
- #processes
- #api
- #forms
- #hex
- #security
- #metaprogramming
Our Sponsors
Build Elixir applications with speed and confidence.
We build reliable cloud platforms for business-critical systems.
Catch errors, track performance, monitor hosts and more.
Error tracking, logs, uptime, and Just Enough APM. Built for developers.
Elixir, Erlang & BEAM conferences from Code Sync - join us in 2026!
Practical resources that improve the lives of professional developers.
Master modern tools, write cleaner code, and tackle real-world challenges.
The deployment platform built for Elixir: PaaS ease, VPS control.
The team behind Membrane, Popcorn, LiveDebugger. Available for hire.
Courses that'll move you from confusion to "Aha, now I get it!"
Showing Posts 1 to 3- Show Best Posts
- Show All (oldest first)
- Show All (newest first)
NduatiK
Have a look at this post https://forum.elixirforum.com/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=cudaAt 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
jan-swiatek
Thanks! I’ll definitely try that.
If I understood correctly, even EXLA on CPU should run faster than training on CPU / GPU?
NduatiK
Yes should be faster than plain Elixir on CPU. Using GPU would be ideal, but for smaller datasets, XLA CPU is pretty fast.
You can find some stats here (just find the word benchmarks)