美文网首页
2023-03-01 在wsl2中运行PyTorch在GPU上训

2023-03-01 在wsl2中运行PyTorch在GPU上训

作者: 胜果铺子 | 来源:发表于2023-02-28 00:57 被阅读0次

遇到的错误提示:

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论坛上的解答

相关文章

网友评论

      本文标题:2023-03-01 在wsl2中运行PyTorch在GPU上训

      本文链接:https://www.haomeiwen.com/subject/vwgkldtx.html