CUDA 12.2 support

[lambdaofgod]

CUDA 12.2 support

[seanmor5]

Can you search for libdevice on /usr/lib/cuda or elsewhere? If you do a find on your system for libdevice and if you find it, then add that directory to:

XLA_FLAGS=--xla_gpu_cuda_data_dir=/usr/lib/cuda

as a system environment variable before running the livebook and see if it fixes your issue

[lambdaofgod]

Yes, it works. Isn't there another way to set this though? I tried passing it to config (adding a line XLA_FLAGS: "--xla_gpu_cuda_data_dir=/usr/lib/cuda") and it doesn't see it.

[jonatanklosko]

@lambdaofgod you can use System.put_env/2 at the very beginning, or Mix.install([...], system_env: %{"XLA_FLAGS" => "--xla_gpu_cuda_data_dir=/usr/lib/cuda"}), or better yet set it as global env var in Livebook settings, or set it in your .bashrc config.