Uehwan / SimVODIS

Simultaneous Visual Odometry, Object Detection, and Instance Segmentation
150 stars 35 forks source link

Cuda error while evaluating the pre-trained model #6

Open sreejithpai opened 2 years ago

sreejithpai commented 2 years ago

Hello, Thank you very much for sharing the implementation. I am trying to run 'evaluate_depth_custom.py' using google colab. I am getting a cuda error " /content/SimVODIS/maskrcnn_benchmark/layers/roi_align.py in forward(ctx, input, roi, output_size, spatial_scale, sampling_ratio) 18 ctx.input_shape = input.size() 19 output = _C.roi_align_forward( ---> 20 input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio 21 ) 22 return output

RuntimeError: cuda runtime error (209) : unrecognized error code at /content/SimVODIS/maskrcnn_benchmark/csrc/cuda/ROIAlign_cuda.cu:297 " Is this due to Cuda version mismatch ? Should I downgrade cuda version in google colab from 11.0 to 9.0 ?

Thanks, Sreejith

sreejithpai commented 2 years ago

Error was resolved after downgrading CUDA version to 10