Ubuntu20.04安装darknet

环境:
Ubuntu20.04
CUDA11.1

darknet是一种类似于tensorflow,pytorch这种深度学习框架,也是一种特征提取器(类似于resnet)
安装darknet主要是想运行yolo(yolo的官方源码是基于darknet写的)

安装步骤:
首先克隆darknet库:在终端中输入git clone https://github.com/pjreddie/darknet.git
然后打开这个文件夹cd darknet
编译make
编译结束后,输入./darknet,如果出现usage: ./darknet 说明安装成功。

如果电脑中有CUDA,那么需要修改darknet文件夹中Makefile文件,把GPU=0 改成 GPU=1。然后remake

在使用OPENCV编译的时候,先在Makefile里将OPENCV=0改成=1,但是按照官方文件https://pjreddie.com/darknet/install/可能会出错,这时候应该去github上下载opencv官方源码git clone https://github.com/opencv/opencv.git,然后

cd opencv
mkdir build
cd build

cmake -D CMAKE_BUILD_TYPE=RELEASE \
      -D CMAKE_INSTALL_PREFIX=/usr/local \
      -D INSTALL_C_EXAMPLES=ON \
      -D INSTALL_PYTHON_EXAMPLES=ON \
      -D WITH_V4L=ON \
      -D WITH_QT=ON \
      -D WITH_OPENGL=ON \
      -D WITH_FFMPEG=ON \
      -D BUILD_EXAMPLES=ON ..

make -j4
sudo make install
sudo sh -c 'echo "/usr/local/lib" >> /etc/ld.so.conf.d/opencv.conf'
sudo ldconfig

然后remake

最后安装mish-cuda

git clone https://github.com/thomasbrandon/mish-cuda
cd mish-cuda
cp external/CUDAApplyUtils.cuh csrc  # cuda >= 11.0
python setup.py build install

你可能感兴趣的:(配置深度学习环境过程记录,目标检测)