Install PyTorch YOLOv5 With CUDA On Ubuntu 20.04
Today we will try to build our environment to host PyTorch YOLOv5 You Only Look Once, The most famous real-time object detection algorithm library with the Nvidia CUDA Driver support.
We will need to do
Update Our Ubuntu 20.04
Install Python And PIP
Nvidia CUDA Driver with CUDA Toolkit version 11.3 for GPU Computing
Nvidia cuDNN The Deep Nural Network Library version libcudnn8_8.2.0.53-1+cuda11.3
Install Pytorch version 1.11.0 stable
Install YOLOv5