Could not load library libcudnn_cnn_train.so.8. Error: /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8: undefined symbol: _ZN5cudnn3cnn34layerNormFwd_execute_internal_implERKNS_7backend11VariantPackEP11CUstream_stRNS0_18LayerNormFwdParamsERKNS1_20NormForwardOperationEmb, version libcudnn_cnn_infer.so.8
cudnn有两个源头,一个是你专门装一个cudnn,另一个是安装pytorch的时候会附带安装nvidia_cudnn***的whl库。
在我这里,这个报错问题不是cudnn的冲突问题,也不是库版本不匹配的问题,而是anaconda下已经安装好的nvidia_cudnn***的库的so文件找不到,所以这个报错的解决方法就是:
export LD_LIBRARY_PATH=/home/aistudio/work/anaconda3/lib/python3.10/site-packages/nvidia/cudnn/lib:$LD_LIBRARY_PATH