1. 程式人生 > >yolo3訓練自己資料時出現cuda error

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 
--------------------- 
作者:一個新新的小白 
來源:CSDN 
原文:https://blog.csdn.net/qq_31511955/article/details/82656353 
版權宣告:本文為博主原創文章,轉載請附上博文連結!

 

 

以及

https://blog.csdn.net/john_bh/article/details/80625220