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.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=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
Last Post!
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)
Popular in Questions
Other popular topics
Latest on Elixir Forum
Sponsor Spotlight
Catch errors, track performance, monitor hosts and more.
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
- #channels
- #elixirconf
- #exunit
- #discussion
- #code-sync
- #javascript
- #podcasts
- #onsite
- #dialyzer
- #docker
- #authentication
- #umbrella
- #full-time-contract
- #podcasts-by-brainlid
- #ecto-query
- #elixir-ls
- #phoenix_html
- #iex
- #blog-post
- #graphql
- #genstage
- #ai
- #websockets
- #supervisor
- #elixirconf-us
- #advent-of-code
- #distillery
- #processes
- #forms
- #api
- #metaprogramming
- #security
- #hex
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.
Error tracking for Elixir devs who love to ship. Start your free account.
Practical resources that improve the lives of professional developers.
Develop your skills with books, videos, and courses.
Producing high quality Elixir screencasts since 2017.
The team behind Membrane, Popcorn, LiveDebugger. Available for hire.
Courses that'll move you from confusion to "Aha, now I get it!"