yuliangguo / Pytorch_Generalized_3D_Lane_Detection

[ECCV 2020] Official PyTorch implementation of 'Gen-LaneNet: a generalized and scalable approach for 3D lane detection'
307 stars 78 forks source link

cuDNN error #25

Open spcrobocar opened 2 years ago

spcrobocar commented 2 years ago

When I run main_demo_GenLanezNet_ext.py, I got the following error:

cuDNN error: CUDNN_STATUS_EXECUTION_FAILED Traceback (most recent call last):
File "main_demo_GenLaneNet_ext.py", line 147, in
output_geo = output_geo[0].data.cpu().numpy() NameError: name 'output_geo' is not defined

How can I fix it?