Installing CUDA

Ubuntu (20.04)

First you want to remove of any remnants of previous driver or CUDA installations:

sudo apt-get --purge -y remove "*cublas*" "cuda*" "nsight*"
sudo apt-get --purge -y remove "*cublas*" "*cufft*" "*curand*" 
sudo apt-get --purge -y remove "*cusolver*" "*cusparse*" "*npp*" "*nvjpeg*"  
sudo apt-get --purge -y remove "nvidia*"

sudo apt-get -y autoremove
sudo apt-get -y autoclean
sudo rm -rf /usr/local/cuda*
sudo apt-get -y update 

Next you can follow the instructions here and do the following to install CUDA 11.2 on Ubuntu 20.04:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda

If you are unsure about your Ubuntu version, check it with:

lsb_release -a

OSX

That won't work... No support for CUDA in recent versions of OSX. However, dexp will still work but be very slow whenever the command or function called is processing intensive. For simple things like copying and cropping datasets, that won't be a problem.

Windows

TODO