【发布时间】:2022-06-27 21:14:39
【问题描述】:
我正在尝试将 Tensorflow 2.7.0 与 GPU 一起使用,但我经常遇到同样的问题:
2022-02-03 08:32:31.822484: W 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; LD_LIBRARY_PATH: /home/username/.cache/pypoetry/virtualenvs/poetry_env/lib/python3.7/site-packages/cv2/../../lib64:/home/username/miniconda3/envs/project/lib/
2022-02-03 08:32:31.822528: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
此问题已在此处和 github 上多次出现。但是,通常建议的解决方案是 a) 下载丢失的 CUDA 文件,b) 降级/升级到正确的 CUDA 版本,c) 设置正确的LD_LIBRARY_PATH。
我已经在我的 PC 上使用了支持 CUDA 的 PyTorch,我没有遇到任何问题。我的nvidia-smi 返回 11.0 版本,这正是我想要的唯一版本。另外,如果我尝试运行:
import os
LD_LIBRARY_PATH = '/home/username/miniconda3/envs/project/lib/'
print(os.path.exists(os.path.join(LD_LIBRARY_PATH, "libcudart.so.11.0")))
它返回True。这正是错误消息中LD_LIBRARY_PATH 的一部分,显然,Tensorflow 看不到libcudart.so.11.0(在那里)。
有什么东西很明显是我遗漏的吗?
nvidia-smi 输出:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.156.00 Driver Version: 450.156.00 CUDA Version: 11.0 |
|-------------------------------+----------------------+----------------------+
nvcc:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243
【问题讨论】:
标签: tensorflow