I'm getting a similar error but my Cudnn is getting internal error. did you find a way to fix your problem? maybe it will work for me aswell.
App Crash Log Traceback (most recent call last):
File "my_design.py", line 61, in run
File "my_DAIN_class.py", line 1538, in RenderVideo
File "my_DAIN_class.py", line 1376, in StepRenderInterpolation
File "my_DAIN_class.py", line 1031, in DoInterpolation
File "my_DAIN_class.py", line 673, in interpolate
File "site-packages\torch\nn\modules\module.py", line 550, in __call__
File "networks\DAIN.py", line 415, in forward
File "site-packages\torch\nn\modules\module.py", line 550, in __call__
File "Resblock\BasicBlock.py", line 81, in forward
File "site-packages\torch\nn\modules\module.py", line 550, in __call__
File "site-packages\torch\nn\modules\container.py", line 100, in forward
File "site-packages\torch\nn\modules\module.py", line 550, in __call__
File "site-packages\torch\nn\modules\conv.py", line 349, in forward
File "site-packages\torch\nn\modules\conv.py", line 345, in _conv_forward
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR