“导入错误：libcupti.so.11.7：无法打开共享对象文件：没有这样的文件或目录”仅适用于 Poetry envs

Question

我正在开发一个使用句子转换器 2.2.2 的项目，如果我使用

创建虚拟环境

python -m venv venv
source venv/bin/activate
pip install -r requirements.txt

应用程序运行没有任何问题。

我正在将项目转换为 Poetry，但是当我使用 Poetry 创建虚拟环境并运行应用程序时，出现此错误：

Traceback (most recent call last):
  File "app/main.py", line 12, in <module>
    from app.src.common.utils import get_categories, get_models
  File "/home/mdonato/Documents-backup/jobs-data-analysis/app/src/common/utils.py", line 10, in <module>
    from sentence_transformers import SentenceTransformer
  File "/home/mdonato/Documents-backup/jobs-data-analysis/.venv/lib/python3.8/site-packages/sentence_transformers/__init__.py", line 3, in <module>
    from .datasets import SentencesDataset, ParallelSentencesDataset
  File "/home/mdonato/Documents-backup/jobs-data-analysis/.venv/lib/python3.8/site-packages/sentence_transformers/datasets/__init__.py", line 1, in <module>
    from .DenoisingAutoEncoderDataset import DenoisingAutoEncoderDataset
  File "/home/mdonato/Documents-backup/jobs-data-analysis/.venv/lib/python3.8/site-packages/sentence_transformers/datasets/DenoisingAutoEncoderDataset.py", line 1, in <module>
    from torch.utils.data import Dataset
  File "/home/mdonato/Documents-backup/jobs-data-analysis/.venv/lib/python3.8/site-packages/torch/__init__.py", line 229, in <module>
    from torch._C import *  # noqa: F403
ImportError: libcupti.so.11.7: cannot open shared object file: No such file or directory

我按照this指南了解 NVIDIA 驱动程序、CUDA 和 cuDNN 的说明。 我也遵循了 Tar 文件安装，所以我尝试以两种方式安装，但我仍然遇到相同的错误。

Answer 1

有一个类似的问题here已解决修复

torch<2.0.1

。在构建 Nvidia 基础镜像的推理 docker 镜像时，同样的解决方案也适用于我。

您的套餐对于火炬版本没有上限。您可能正在安装

2.0.1

，这也给了我您所看到的错误。