第一個問題:
CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0' failed. 已放棄 (核心已轉儲)
由Error類型可以知道運行模型所需的顯存空間超過了顯卡提供的顯存。
解決辦法:
修改cfg配置文件,減小batch和subdivsion.
第二個問題:
CUDA Error: unknown error darknet: ./src/cuda.c:36: check_error: Assertion `0' failed. Aborted
這個問題的發生,網上給出了很多原因,可以嘗試解決下。
我記錄下我這塊的原因:我這是由於nvidia驅動版本和電腦內核支持版本不統一造成的。
如何確定是不是這個這個原因,可通過在終端輸入 nvidia-smi,顯示如下:
Failed to initialize NVML: driver/library version mismatch
解決方法可參看如下博客,寫的很好:
https://blog.csdn.net/zhang970187013/article/details/81012845
另外,我的是支持版本是384的,下載地址為:
https://www.geforce.cn/drivers/results/133208