Check failed: status == CUBLAS_STATUS_SUCCESS (13 vs. 0) CUBLAS_STATUS_EXECUTION_FAILED (Memory leak)

[PofeiShit]

您好，很感谢您的代码，迁移效果非常好。但是会有如下报错
F1115 11:06:27.330569 9959 math_functions.cu:85] Check failed: status == CUBLAS_STATUS_SUCCESS (13 vs. 0) CUBLAS_STATUS_EXECUTION_FAILED
*** Check failure stack trace: ***
Aborted (core dumped)
我重复了几次都会出现这个错误，我观察了下每次都是跑了39张图崩溃, GPU显存从一开的4G慢慢增加到16G满了。不知道是否有内存泄漏问题。

[PofeiShit]

这里添加cudaFree(data_C[i]);

Neural-Color-Transfer/code/windows/neural_color_transfer/source/main.cu

Line 447 in eb86d93

cudaFree(data_S1[i]);