This gist contains instructions about cuda v11.1 and cudnn 8.0.4 installation in Ubuntu 18.04 for Tensorflow 2.4.1
- If you have previous installation remove it first.
!/bin/bash
sudo apt-get purge *nvidia*
sudo apt remove --autoremove nvidia-*
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove --autoremove nvidia-cuda-toolkit
sudo apt-get autoremove && sudo apt-get autoclean