RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

michaildoukas / head2head

PyTorch implementation for Head2Head and Head2Head++. It can be used to fully transfer the head pose, facial expression and eye movements from a source video to a target identity.

MIT License

310 stars 44 forks source link

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED #19

Closed cyrala closed 2 years ago

cyrala commented 2 years ago

Hi, I installed environment according to conda-env.txt（python==3.7.6, pytorch==1.4.0, cuda==9.2）, but when I run code, it shows' RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED' , can you give me some advice? Thank you!