Использование RTX3090 с Haystack

Question

Текущая версия Haystack (deepset.ai) поддерживает графические процессоры, но не использует достаточно новой поддержки CUDA (>= 11.1) для работы с моим RTX3090. Могу я просто удалить текущую версию torch и переустановить с поддержкой cu111? Или другие компоненты также привязаны к более ранней версии библиотеки CUDA?

Answer 1

У меня установлен torch 1.9.0 с установленным CUDA 11.2, и он отлично работает с основными ветвями FARM и haystack. Позвольте мне рассказать вам, как туда добраться.

Проблема в том, что последняя версия haystack 0.9.0 имеет FARM 0.8.0 в качестве зависимости, которая исправляет зависимость torch до <1.9. Однако, если вы устанавливаете haystack из его основной ветки, вы можете удалить фиксированную зависимость FARM 0.8.0 и установить FARM также из основной ветки. На текущем главном ответвлении FARM зависимость от факела ослаблена до <1,10.

Я имею в виду следующие шаги, чтобы сначала установить 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 .

Обратите внимание, что все это не понадобится, когда будет выпущена версия стога сена 1.0.0, которая будет в течение следующих 3 недель в конце сентября 2021 года. Возможно, еще до этого будет выпущена незначительная версия 0.9.1, которая должна решите свою проблему, чтобы вы могли просто использовать .