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

Question

我正在尝试在我的 Django Rest 框架项目中导入和使用 ultralytics 库，我使用诗歌作为我的依赖项管理器，我使用安装了 ultralytics，poetry add ultralytics并在尝试在我的代码中导入该库时收到此错误

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

我该如何解决这个问题？

Answer 1

更新：此问题已在 PyTorch 2.1.1 中修复。

自 2023 年 5 月 9 日起，PyTorch 2.0.1 和 2.1.0存在一个未解决的问题，导致poetry lock从poetry.lock. 他们的wheel包含依赖项，但是他们的PyPi上传没有得到它。

解决方法是在 pyproject.toml 中跳过这些版本（通常在 下[tool.poetry.dependencies]）：

torch = ">=2.0.0, !=2.0.1, !=2.1.0"

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

poetry lock --no-update
poetry install

Answer 2

您必须在系统上安装 cuda。

例如在ubuntu上：

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