问题描述
在运行pytorch代码时,遇到如下的报错信息:
Could not load library libcudnn_cnn_train.so.8. Error: /usr/local/cuda/lib64/libcudnn_cnn_train.so.8: undefined symbol
RuntimeError: GET was unable to find an engine to execute this computation
解决方法
参考https://github.com/ultralytics/.github/issues/43
(1)打开终端,echo $LD_LIBRARY_PATH,并进入到打印出的目录;
(2)重命名 libcudnn_cnn_train.so.8为一个副本;
(3)现在系统就不会使用该目录下的 libcudnn_cnn_train.so.8,而是使用pytorch自带的库。