RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

[Serena233]

hello~I tried to run your code, but this showed up:RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS, and I search online, someone told me it maybe because there is something wrong with the environment.
I have installed pytorch0.4.0 cuda 8.0 cudnn7.0.5 torchvision0.2.0. Could you please tell me your environment? Thanks a lot.

[chmxu]

Hi, unfortunately I cannot remember the exact environment I used when conducting this research. But you can first create an environment that is compatible with Pytorch0.4. I think such an environment works for this project.