实验测试darknet,遇到错误CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.
)
一、问题描述
实验测试darknet,遇到错误CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0’ failed.
二、首先需要修改Makefile文件
修改一
GPU = 1
CUDNN = 1
OPENCV = 0 # 根据自己需要修改opencv
修改二
ARCH= -gencode arch