RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

[yushz]

Hi, thanks for your excellent code and paper.I am reproducing this paper using my own dataset, but I am getting this error.The pytorch version is the correct one.

[che1007]

Did you use multiple gpu? I also met the similar issue. How did you solve it?