【发布时间】:2021-08-23 18:46:30
【问题描述】:
我刚刚将 TF 安装到一个全新的 conda 环境中,并尝试安装几个版本的 cudnn,但不确定是否相关。 GPU 无法工作,因为我收到此错误:
tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory
确实,我查看了/usr/local/cuda/lob64,我看到的文件是libcudart.so.10.1,所以我猜它不是正确的版本(10 应该是11)。有谁知道我需要安装哪个软件包才能获得这个libcudart.so.11.0 文件?如果不是,也许我可以使用旧版本的 tensorflow? (我只是尝试pip install tensorflow==2.3.0 无济于事,找不到包。)所以我认为更好的选择是以某种方式获取更新的 libcudart 文件,也许这意味着我需要升级 cuda?
【问题讨论】:
-
好吧,我直接用 conda 而不是 pip 安装了,不知怎的,它想出了如何安装正确的版本并且它可以工作!
-
如果你使用 condos 安装所有东西,你应该得到他们的 cuda-toolkit 包的匹配版本,它将为你安装的 tensorflow 版本提供正确版本的运行时库。然而,你已经设法打破了这一点,你显然需要 CUDA 11.0 运行时
-
@talonmies 是的,谢谢,在我遗漏的问题中,我使用 pip 安装,因为这些是 TF 网站上的官方说明。但是,我找到了一个教程,该教程建议您使用 conda 进行安装,是的,它可以工作。
标签: python tensorflow gpu