PyTorch remains so many bugs, feeling tired to use this framework…
So, the problem is, I compiled pytorch from source, and set CUDNN to my own path,
there is nowhere exist another CUDNN or CUDA.
In the building log, cuDNN version : 7.1.4, but when I run lstm, it given following error:
RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7104 but linked against 7005.
So, can anyone help me ??