https://hub.docker.com/r/kimnt93/nvidia-faiss-python
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
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, defaultBUILD_CUDA_VERSION=10(see available tags at https://hub.docker.com/r/nvidia/cuda/) -
FAISS_VERSION: version offaisshttps://github.com/facebookresearch/faiss -
CC: compute capability based on your GPU product. To know yourccplease visit this link. You must remove thedot, example GeForce GTX 1060 has compute capability=7.5, soCC=6.1
If you want to build docker for Geforce RTX 2080 Ti model, run the command below:
nvidia-docker build --build-arg CC=75 .
MIT