During the YOLOv3 deep learning setup, have to make sure that your nvidia-smi command is working fine and proceed to the next leve,
$ sudo nvidia-smi
Open your darknet "Makefile" and check the configuration looks like below, If you did not set the nvcc in your $PATH can set the direct location of nvcc bin file,
GPU=1
CUDNN=1
OPENCV=0
CC=gcc
CPP=g++
NVCC=nvcc
Build the darknet,
$ sudo make
Choose the number of GPUs based on which are available in your machine, I have 2 GPUs in my machine.
$sudo ./darknet detector train model.data yolov3.cfg darknet53.conv.74 -gpus 0,1
Comments (0)