将 RTX3090 与 Haystack 一起使用答案

【问题标题】：Using RTX3090 with Haystack将 RTX3090 与 Haystack 一起使用
【发布时间】：2021-10-30 13:15:10
【问题描述】：

当前版本的 Haystack (deepset.ai) 支持 GPU，但它没有使用足够新的 CUDA 支持 (>= 11.1) 来与我的 RTX3090 一起工作。我可以卸载当前版本的 torch 并重新安装支持 cu111 吗？或者其他组件是否也与早期版本的 CUDA 库相关联？

【问题讨论】：

一般来说，您不能在一个单一的可执行文件中混合依赖于不同版本的 CUDA 运行时 API 的二进制文件。但这是否适用于您在问题中引用的任何 NLP 内容取决于您在问题中引用的 NLP 内容，而不是 CUDA 本身

标签： nlp gpu nvidia torch haystack

【解决方案1】：

我安装了安装了 CUDA 11.2 的 torch 1.9.0，它可以与 FARM 和 haystack 的主分支一起正常工作。让我带您了解如何到达那里的步骤。

问题是 haystack 0.9.0 的最新版本具有 FARM 0.8.0 作为依赖项，这将 torch 依赖项修复为

我想到了以下步骤，首先安装 FARM，然后从他们的主分支安装 haystack：

  git clone https://github.com/deepset-ai/FARM.git
  cd FARM
  pip install -r requirements.txt
  pip install --editable .
  cd ..
  git clone https://github.com/deepset-ai/haystack.git
  cd haystack

  #now edit requirements.txt and remove line with farm==0.8.0

  pip install --editable .

请注意，在 2021 年 9 月下旬的未来 3 周内发布 1.0.0 版 haystack 时，所有这些都不是必需的。也许在此之前还会有一个次要的 0.9.1，它应该可以解决您的问题，以便您可以简单地使用pip install farm-haystack。

【讨论】：