基于darknet 的yolov4的Ubuntu 18.04环境搭建

1.Nvidia 驱动的下载安装
可以通过手动下载驱动的方式安装

卸载之前安装的驱动sudo apt-get purge nvidia*

禁用nouveau驱动
sudo vim /etc/modprobe.d/blacklist-nouveau.conf

写入
blacklist nouveau
options nouveau modeset=0

更新
sudo update-initramfs -u

重启 打开命令行模式

关闭桌面模式 
sudo service lightdm stop

安装 gcc与g++,18.04需要安装
sudo apt-get install g++

chmod a+x NVIDIA***.run
sudo./NVIDIA***.run -no-opengl-files

开启桌面服务 sudo service lightdm start

查看驱动 watch -n  0.1 nvidia-smi

2.Cuda与cudnn的安装

下载安装cuda,取消安装驱动,设置环境变量
sudo gedit ~/.bahrc
写入
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

source ~/.bashrc

cudnn 解压文件,复制对应得文件到 /usr/local/cuda-11.0(安装的版本号)下

3.安装opecnv

sudo apt-get install libopencv-dev

4.更改Makefile 文件

    GPU=1
	CUDNN=1
	CUDNN_HALF=1
	OPENCV=1
	LIBSO=1
	ARCH  计算能力改为对应的显卡代码  -gencode 61  //1080ti是这个

5,进入Darknet文件夹

make -j8
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值