遇到的错误提示：

Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory

解决尝试一[失败]

1. 先找到这个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论坛上的解答