Posted on Thu 05 December 2019

Installing CUDA on Ubuntu 19.10

I just installed CUDA on my Ubuntu 19.10 desktop, and it was a pretty quick process, if non-obivous.

  1. Installed base CUDA with apt install nvidia-cuda-toolkit
  2. Downloaded libcudnn from [cached]nvidia and installed with dpkg -i. (I couldn't find it in the official Ubuntu repo, let me know if it is there)
  3. Set XLA_FLAGS=--xla_gpu_cuda_data_dir=/usr/lib/cuda/ so JAX can find the installed libraries.

Tags: misc

© Julian Schrittwieser. Built using Pelican. Theme by Giulio Fidente on github. .