环境:
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 <function>
说明安装成功。
如果电脑中有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