ubuntu18环境搭建Darknet4环境

ubuntu18下RTX3070+cuda11.1+opencv4.2.0+darknet环境搭建

整体安装过程分为"5"个步骤

step1:RTX3070显卡安装

一、安装前准备

创建文件：

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

在文件内插入以下内容：

blacklist nouveau
options nouveau modeset=0

执行以下命令使禁用生效并且重启：

sudo update-initramfs -u
sudo reboot

重启后验证是否生效：

lsmod | grep nouveau

没有输出，则禁用生效。

二、安装

卸载原有的显卡驱动

sudo apt-get remove --purge nvidia-*  
sudo apt-get autoremove
sudo reboot

方法1：

1、通过添加ppa源安装驱动

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

2、输入下列并按两下tab，可看出哪些驱动可被安装

sudo apt-get install nvidia-

3、具体安装哪个版本自行选择,此处选择

sudo apt-get install nvidia-driver-455

方法2
所以采用此方法安装最新显卡驱动

1、驱动下载

2、安装准备

卸载之前显卡驱动

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo reboot

3、尝试运行安装

sudo ./NVIDIA-Linux-x86_64-455.38.run

安装过程

选“Yes”

后面过程都直接继续下一步，直至安装完成。
安装完成后输入

nvidia-smi

安装过程中如提示如下错误
则执行下列操作后重新尝试安装

sudo su
systemctl isolate multi-user.target
modprobe -r nvidia-drm

step2:安装cuda

cuda11.1下载官网，根据选择
https://developer.nvidia.com/zh-cn/cuda-downloads

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

直接在ubuntu上下载比较慢，一个小技巧，使用迅雷下载将会很快。
下载完后安装，根据提示
1、输入 accept
2、去掉 driver选项，不进行显卡驱动安装，其余保留
3、一直往下直至安装完成
在~/.bashrc中添加一下环境变量时cuda生效

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda
export LIBRARY_PATH=/usr/local/cuda/lib64:$LIBRARY_PATH

source ~/.bashrc;
nvcc --version

确定cuda安装成功

step3:安装cudnn

根据cuda版本到官网下载cudnn版本，下载过程需要注册、登录、人机验证，比较繁琐
https://developer.nvidia.com/rdp/cudnn-archive

然后执行

tar cudnn-11.1-linux-x64-v8.0.4.30.tgz                                                                                                                                 
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/* /usr/local/cuda/lib64/

step4:安装opencv

查看另一篇文章
https://blog.youkuaiyun.com/gwc_master/article/details/109545042

step5:安装darknet4

注：之前下载过darknet4的可能不支持cuda 11.1，最终效果就是darknet4编译报错，确定是否支持。
1、根据GPU算力修改ARCH:
https://developer.nvidia.com/zh-cn/cuda-gpus
查看自身GPU型号的算力，再根据自身GPU算力查看Makefile中是否有支持

如果支持80算力的显卡则支持RTX3070，否则更新darknet4
https://github.com/AlexeyAB/darknet

2、修改这前四处；

make -j16；
image_yolov4.sh

如果可成功加载网络说明环境正常；
下载权重文件
https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights