yolo3训练自己数据时出现cuda error

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 
--------------------- 
作者：一个新新的小白 
来源：优快云 
原文：https://blog.youkuaiyun.com/qq_31511955/article/details/82656353 
版权声明：本文为博主原创文章，转载请附上博文链接！

 

 

以及

https://blog.youkuaiyun.com/john_bh/article/details/80625220