ValueError：在系统路径中找不到 libcublas.so.*[0-9]

Question

我正在尝试在 Django Rest 框架项目中导入和使用 ultralytics 库，我使用诗歌作为我的依赖管理器，我使用

poetry add ultralytics

安装了 ultralytics，并在尝试在我的代码中导入该库时收到此错误

ValueError: libcublas.so.*[0-9] not found in the system path [my project and virtual environment paths]

我该如何解决这个问题？

Answer 1

自 2023 年 5 月 9 日起，PyTorch 2.0.1 存在一个未解决的问题，导致

poetry lock

从

poetry.lock

删除 libcublas。他们的wheel包含依赖项，但是他们的PyPi上传没有得到它。

解决方法是在 pyproject.toml 中跳过此版本：

torch = ">=2.0.0, !=2.0.1"

确保在进行更改后运行以下命令以正确更新您的诗歌环境

poetry lock --no-update
poetry install

Answer 2

您必须在系统上安装cuda。

例如在 ubuntu 上：

sudo apt-get install nvidia-cuda-toolkit nvidia-cuda-toolkit-gcc

Answer 3

如果您来到这里并在使用诗歌时遇到其他一些库的问题，请尝试直接将其安装在

.venv

中。对我来说是：

   pip3 install "transformers[torch]"