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