kimnt93/nvidia-faiss-python repository overview

⁠Docker Faiss and Python3

https://github.com/userkimcs/faiss-docker⁠

Faiss is a library for vectors search developed by Facebook AI Research. For more information on Faiss please visit https://github.com/facebookresearch/faiss⁠

To start container with Python interface run

nvidia-docker run kimnt93/nvidia-faiss-python

⁠Build

The lasted version run on GeForce GTX 1060 with compute capability=6.1. You can build your own image

nvidia-docker build --build-arg <ARG> .

with arguments

• BUILD_CUDA_VERSION: cuda version, default BUILD_CUDA_VERSION=10 (see available tags at https://hub.docker.com/r/nvidia/cuda/⁠)

• FAISS_VERSION: version of faiss https://github.com/facebookresearch/faiss⁠

• CC: compute capability based on your GPU product. To know your cc please visit this link⁠. You must remove the dot, example GeForce GTX 1060 has compute capability=7.5, set CC=6.1

⁠Example

If you want to build docker for Geforce RTX 2080 Ti model, run the command below:

nvidia-docker build --build-arg CC=75 .

⁠License

MIT