mirror of
https://github.com/facebookresearch/faiss.git
synced 2026-10-11 22:50:00 +00:00
Ikko Eltociear Ashimine
93cc1d7925
Summary: necesary -> necessary Pull Request resolved: https://github.com/facebookresearch/faiss/pull/2648 Reviewed By: alexanderguzhva Differential Revision: D42805978 Pulled By: mlomeli1 fbshipit-source-id: 1d640918dfe85e79856b18ce718fb26bbb0c6a92
30 lines
1.1 KiB
Docker
30 lines
1.1 KiB
Docker
FROM nvidia/cuda:8.0-devel-centos7
|
|
|
|
# Install MKL
|
|
RUN yum-config-manager --add-repo https://yum.repos.intel.com/mkl/setup/intel-mkl.repo
|
|
RUN rpm --import https://yum.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB
|
|
RUN yum install -y intel-mkl-2019.3-062
|
|
ENV LD_LIBRARY_PATH /opt/intel/mkl/lib/intel64:$LD_LIBRARY_PATH
|
|
ENV LIBRARY_PATH /opt/intel/mkl/lib/intel64:$LIBRARY_PATH
|
|
ENV LD_PRELOAD /usr/lib64/libgomp.so.1:/opt/intel/mkl/lib/intel64/libmkl_def.so:\
|
|
/opt/intel/mkl/lib/intel64/libmkl_avx2.so:/opt/intel/mkl/lib/intel64/libmkl_core.so:\
|
|
/opt/intel/mkl/lib/intel64/libmkl_intel_lp64.so:/opt/intel/mkl/lib/intel64/libmkl_gnu_thread.so
|
|
|
|
# Install necessary build tools
|
|
RUN yum install -y gcc-c++ make swig3
|
|
|
|
# Install necessary headers/libs
|
|
RUN yum install -y python-devel numpy
|
|
|
|
COPY . /opt/faiss
|
|
|
|
WORKDIR /opt/faiss
|
|
|
|
# --with-cuda=/usr/local/cuda-8.0
|
|
RUN ./configure --prefix=/usr --libdir=/usr/lib64 --without-cuda
|
|
RUN make -j $(nproc)
|
|
RUN make -C python
|
|
RUN make test
|
|
RUN make install
|
|
RUN make -C demos demo_ivfpq_indexing && ./demos/demo_ivfpq_indexing
|