博客作者遇到一个问题，即符号链接cuda-11.1的库文件到tensorflow的site-packages未成功，但直接将库文件放入tensorflow的site-packages目录解决了问题。这个问题可能与CUDA版本和TensorFlow的兼容性有关。作者建议遇到类似问题的人查看Hoefling在#43947评论中提供的额外详情。
For some reason the symlink did not work with the provided target, however placing it in the tensorflow site-packages worked properly
sudo ln -s /usr/local/cuda-11.1/targets/x86_64-linux/lib/libcusolver.so.11 /home/YourUsernameHere/.local/lib/python3.8/site-packages/tensorflow/python/libcusolver.so.10
(Obviously replacing python3.8 with proper value as well as username)
Edit: For those finding their way here, I recommend seeing the extra details provided by hoefling below (#43947 (comment))
扫一扫
840
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
