cuda error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0' failed.
解决方法:
一、Darknet编译使用GPU要求显卡是Nvidia卡并且正确安装了CUDA。
GPU环境下的编译配置都是在 /darknet/Makefile 文件中定义的,GPU环境的编译有3点更改需要注意。
1. 更改Makefile前两行GPU和CUDNN的配置:
GPU=1
CUDNN=1
2、更改CUDA的路径
48~51行,在”ifeq ((GPU),1)”语句块中修改为自己的CUDA安装路径,更改前默认路径如下:ifeq((GPU),1)”语句块中修改为自己的CUDA安装路径,更改前默认路径如下:ifeq((GPU), 1)
COMMON+= -DGPU -I/usr/local/cuda/include/
CFLAGS+= -DGPU
LDFLAGS+= -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcurand
将三处的cuda路径改为自己本机上的路径即可
修改NVCC的路径:
NVCC=/usr/local/cuda-8.0/bin/nvcc
---------------------
作者:一个新新的小白
来源:CSDN
原文:https://blog.csdn.net/qq_31511955/article/details/82656353
版权声明:本文为博主原创文章,转载请附上博文链接!
以及