遇到的错误提示:
Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory
解决尝试一[失败]
- 先找到这个
libcudnn_cnn_infer.so.8
文件所在的位置[^1]
sudo find / -name 'libcudnn.so.8'
然后export到库文件查找路径中
export LD_LIBRARY_PATH=<PATH_OF_LIBRARY_FROM_ABOVE_CODE>:$LD_LIBRARY_PATH
我找到的路径是/usr/lib/x86_64-linux-gnu
尝试一结果
没有解决
解决尝试二
直接用pytorch论坛[^2]上给的地址,而不是像stackoverflow答案说的自己查找
export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH
结果
成功解决!正常在GPU上训练
参考
[^1] stackoverflow
[^2] pytorch论坛上的解答