Ubuntu下YOLOv5目标检测
软件环境
-
ubuntu 20.4
-
anaconda3
-
cuda 11.1
-
pytorch 1.7
-
YOLOv5
环境安装
1 安装Anaconda
2 安装cuda和cudnn
-
安装Nvidia驱动
确认你的系统已经装好了nvidia显卡驱动,没有的话从nvidia官网下载适配显卡的驱动。
-
安装CUDA
接着安装cuda,打开终端:
wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run sudo sh cuda_11.1.0_455.23.05_linux.run
打开
~/.bashrc
:su