FAISS-GPU не может использоваться с GPU версии 30xx

Question

Я пишу исходный код Python и использую Faiss. Я могу использовать Faiss на CPU, GPU 20xx, например: RTX 2080Ti, ... Однако, когда я запускаю на таких устройствах, как RTX 3060, RTX 3070, система зависает, и я не могу убить программу с помощью Ctrl + C. исходный код, который я использую: https://github.com/facebookresearch/faiss/blob/main/tutorial/python/5-Multiple-GPUs.py

Если кто-то столкнулся с этой ошибкой и успешно ее исправил, поделитесь со мной, как это сделать.

ОС: Ubuntu 20.04

Версия Faiss: выпуск 1.7.1

Работает на: GPU + Python

Answer 1

Я использовал Anaconda, я установил ниже, это сработало. Если вы используете Docker, установите Miniconda.

conda install faiss-gpu cudatoolkit=11.1 -c pytorch-gpu

conda install -c anaconda pytorch-gpu