caffe2 CUDNN_STATUS_NOT_INITIALIZED

时间:2018-07-15 03:05:11

标签: cudnn caffe2

运行Detectron时,caffe2报告CUDNN_STATUS_NOT_INITIALIZED: CUDA版本:9.1 cudnn版本:7.0 os:ubuntu 16

Traceback (most recent call last):
File "tools/train_net.py", line 360, in <module>
main()
File "tools/train_net.py", line 198, in main
checkpoints = train_model()
File "tools/train_net.py", line 219, in train_model
workspace.RunNet(model.net.Proto().name)
File "/mnt/workspace/wqm/caffe2-for-
detectron/pytorch/build/caffe2/python/workspace.py", line 217, in RunNet
StringifyNetName(name), num_iter, allow_fail,
File "/mnt/workspace/wqm/caffe2-for-detectron/pytorch/build/caffe2/python/workspace.py", line 178, in CallWithExceptionIntercept
return func(*args, **kwargs)
RuntimeError: [enforce fail at context_gpu.h:100] status == 
CUDNN_STATUS_SUCCESS. 1 vs 0. , Error at: /mnt/workspace/wqm/caffe2-
for-detectron/pytorch/caffe2/core/context_gpu.h:100: 
CUDNN_STATUS_NOT_INITIALIZED Error from operator: 
input: "gpu_0/data" input: "gpu_0/conv1_w" output: "gpu_0/conv1" name: 
"" type: "Conv" arg { name: "kernel" i: 7 } arg { name: 
"exhaustive_search" i: 0 } arg { name: "pad" i: 3 } arg { name: 
"order" s: "NCHW" } 
arg { name: "stride" i: 2 } device_option { device_type: 1 
cuda_gpu_id: 0 } engine: "CUDNN"   

0 个答案:

没有答案