问题
RuntimeError: cuDNN version incompatibility: PyTorch was compiled against (8, 5, 0) but found runtime version (8, 4, 1). PyTorch already comes bundled with cuDNN. One option to resolving this error is to ensure PyTorch can find the bundled cuDNN.Looks like your LD_LIBRARY_PATH contains incompatible version of cudnnPlease either remove it from the path or install cudnn (8, 5, 0)
在复现memsum,构建环境的时候,出现了这个问题,逛了一圈,问题出在pytorch和cudnn版本不对,可以从LD_LIBRARY_PATH解决问题
方法
unset LD_LIBRARY_PATH
解决了我的问题