Failed building wheel for faiss-gpu

[fire717]

while run pip install gedml

my env:

ubuntu18 python3.8 pytorch1.8.2 cuda11.1

[zbr17]

[Possible cause] Sorry for the late reply. We need to specify the faiss-gpu version as 1.5.3, which may not be compatible with the latest PyTorch and python.

[Solution] It is ok to install gedml without faiss-gpu package. We write the related code in src/gedml/core/evaluators/metrics.py as follows (Line 10-17):

# faiss
try:
    import faiss
except ModuleNotFoundError:
    logging.warning("Faiss Package Not Found! (metrics package)")

_USE_FAISS_ = False
_NORMALIZE_ = True

You need to keep the flag _USE_FAISS_ to be False and delete the faiss-gpu related item in requirements.txt and try to re-install again.

---

You can always contact me if any other problem occurs. 😄