Install CUDA Toolkit for your AI
First thing is to install NVIDIA CUDA Toolkit. but first lets just make sure the packages are up to date.
sudo apt update && sudo apt upgrade
If there were any updates it's a good idea to reboot to make sure there are no errors from updating