install-yolo-tensorflow-gpu-cuda-10-0-cudnn-ubuntu-1804-nvidia-driver-440

################ NVIDIA DRIVER 440 INSTALLATION
# Install NVIDIA driver
wget http://us.download.nvidia.com/XFree86/Linux-x86_64/440.64/NVIDIA-Linux-x86_64-440.64.run
chmod a+x NVIDIA-Linux-x86_64-440.64.run
sudo dpkg --add-architecture i386
sudo apt install lib32gcc1 build-essential dkms
sudo ./NVIDIA-Linux-x86_64-440.64.run --no-cc-version-check
# Reboot. Check that GPUs are visible using the command: nvidia-smi

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

sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/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-0 cuda-tools-10-0 cuda-compiler-10-0 cuda-libraries-10-0 cuda-libraries-dev-10-0

# 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.0_amd64.deb 
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.0_amd64.deb 
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.0_amd64.deb

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

# Download TensorRT 6 from nvidia website:
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.0-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==1.2.0 torchvision==0.4.0

################ 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==1.15

## 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
#