mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
3ac3ca0fab
Summary: This PR paves the way for nightly builds. + Get rid of cmake 3.17 manual install as cmake 3.18 is now available in conda. + Update docker files for conda packages. + Specify CUDA architectures via CMake's `CMAKE_CUDA_ARCHITECTURES`. Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1422 Reviewed By: mdouze Differential Revision: D23870447 Pulled By: beauby fbshipit-source-id: 40ae7517e83356443a007a43261713e7e3a140d4
17 lines
530 B
Docker
17 lines
530 B
Docker
FROM nvidia/cuda:10.2-devel-ubuntu18.04
|
|
|
|
RUN apt-get update && apt-get install -y wget git libcublas-dev
|
|
|
|
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
|
|
bash Miniconda3-latest-Linux-x86_64.sh -b -p ~/miniconda3
|
|
ENV PATH="/root/miniconda3/condabin:${PATH}"
|
|
|
|
RUN conda install conda-build
|
|
|
|
COPY ./ faiss
|
|
WORKDIR /faiss/conda
|
|
|
|
RUN CUDA_ARCHS="35;52;60;61;70;72;75" \
|
|
conda build faiss-gpu --variants '{ "cudatoolkit": "10.2" }' \
|
|
--no-anaconda-upload -c pytorch --no-test
|