install-cuda-10-1-cudnn-ubuntu-2004-nvidia-driver-440-yolo-tensorflow-gpu

gistfile1.txt

################ NVIDIA DRIVER 440 INSTALLATION
# Install NVIDIA driver
# Ubuntu 20.04 LTS already has nvidia driver version 440

################ NVIDIA CUDA 10.0 INSTALLATION

sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
sudo apt-get update
# Here we shouldn't install neither cuda-10-0, cuda-runtime-10-0 nor cuda
# Because these package will install a nVidia Driver on top of our previously installed
sudo apt-get install cuda-toolkit-10-1 cuda-tools-10-1 cuda-compiler-10-1 cuda-libraries-10-1 cuda-libraries-dev-10-1

# add to PATH
echo 'export PATH="$PATH:/usr/local/cuda/bin"' | tee -a ~/.bashrc
source ~/.bashrc

################ NVIDIA CUDNN 7.6.5 INSTALLATION

# Install CUDNN 7.6.5
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.1_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.1_amd64.deb

################ NVIDIA TensorRT INSTALLATION

# Download TensorRT 6 from nvidia website:
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.1-trt6.0.1.5-ga-20190913_1-1_amd64.deb
sudo apt-get update

# Install TensorRT. Requires that libcudnn7 is installed above.
sudo apt-get install -y --no-install-recommends libnvinfer6 \
    libnvinfer-dev \
    libnvinfer-plugin6

################ PyTorch INSTALLATION
################ FROM https://pytorch.org/get-started/locally/
sudo apt install python3-pip python-pip

sudo -H pip3 install torch torchvision

################ TensorFlow INSTALLATION
################ FROM https://github.com/heethesh/Computer-Vision-and-Deep-Learning-Setup/blob/master/README.md#1-install-prerequisites

sudo apt-get install python3-testresources
sudo -H pip3 install --upgrade pip setuptools
sudo pip3 install tensorflow-gpu

## References
# https://github.com/AlexeyAB/darknet
# https://pjreddie.com/darknet/yolo/
# https://github.com/tiagoshibata/darknet
# https://github.com/AlturosDestinations/Alturos.Yolo
# https://medium.com/@cjanze/how-to-install-tensorflow-with-gpu-support-on-ubuntu-18-04-lts-with-cuda-10-nvidia-gpu-312a693744b5
#