RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling XXXX
印象中,我的就是显存不够,使用torch工具nn.DataParallel时报的显存不够,改小batch后正常。未使用且未改小时报了这个不知所云的错
0. 其他参考
秦耘的CSDN博客: 报错:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED.
stack overflow上: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling ‘cublasCreate(handle)’.