search

raghudeep / netwarp_public

Semantic Video CNNs through Representation Warping. ICCV 2017
BSD 3-Clause "New" or "Revised" License
69 stars 20 forks source link

run error about cudnn issue #3

Closed amiltonwong closed 6 years ago

amiltonwong commented 6 years ago

Hi, @raghudeep

When I run: python scripts/run_netwarp.py data_split models/pspnet101_cityscapes_conv5_4netwarp_deploy.prototxt models/pspnet101_cityscapes_conv5_4netwarp.caffemodel results 1

I got the following error:

F1110 21:45:58.179529 28101 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0)  CUDNN_STATUS_INTERNAL_ERROR
*** Check failure stack trace: ***
Aborted (core dumped)

It seems the cudnn issue. I had installed cuda-8.0 and cudnn 6.0 in my system

root@milton-ThinkCentre-M93p:~# cat /usr/local/cuda/version.txt
CUDA Version 8.0.61
root@milton-ThinkCentre-M93p:~# cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR      6
#define CUDNN_MINOR      0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION    (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

Could you give some suggestion to fix this problem?

THX!

amiltonwong commented 6 years ago

I finally found out it was the memory problem. Now it's fixed