运行LeNet出现的错误。请大家帮我分析一下原因?


#1

以下的讨论是基于:
MXNet版本: 1.5.1(gpu)
操作系统: win10
c:\jenkins\workspace\mxnet\mxnet\src\operator\nn\cudnn./cudnn_algoreg-inl.h:97: Running performance tests to find the best convolution algorithm, this can take a while… (setting env variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable)
Traceback (most recent call last):
File “c:/Users/Administrator/Desktop/deeplearning/LeNet.py”, line 29, in
com.train_ch5(net,trainIter,testIter,loss,batchSize,trainer,ctx,numEpoch)
File “c:\Users\Administrator\Desktop\deeplearning\common.py”, line 85, in train_ch5
trainSum += l.asscalar()
File “C:\ProgramData\Anaconda3\lib\site-packages\mxnet\ndarray\ndarray.py”, line 2013, in asscalar
return self.asnumpy()[0]
File “C:\ProgramData\Anaconda3\lib\site-packages\mxnet\ndarray\ndarray.py”, line 1995, in asnumpy
ctypes.c_size_t(data.size)))
File “C:\ProgramData\Anaconda3\lib\site-packages\mxnet\base.py”, line 252, in check_call
raise MXNetError(py_str(_LIB.MXGetLastError()))
mxnet.base.MXNetError: [18:35:28] c:\jenkins\workspace\mxnet\mxnet\src\operator\linalg_impl.h:212: Check failed: e == CUBLAS_STATUS_SUCCESS (13 vs. 0) cuBLAS: CUBLAS_STATUS_EXECUTION_FAILED