cuDNN failed to initialize

google-research / ravens

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.

https://transporternets.github.io

Apache License 2.0

562 stars 97 forks source link

cuDNN failed to initialize #13

Closed MeetGandhiMagna closed 3 years ago

MeetGandhiMagna commented 3 years ago

This is the error that I am getting when trying to run train.py: E tensorflow/stream_executor/cuda/cuda_dnn.cc:328] Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

After googling the error, I tried restarting my PC as well as other steps mentioned; however none helped.

DanielTakeshi commented 3 years ago

@MeetGandhiMagna Can you provide the complete installation steps you performed? Thanks.

MeetGandhiMagna commented 3 years ago

The issue has been resolved after updating Ubuntu.