facebookresearch/faiss
1
0
Fork 0
mirror of https://github.com/facebookresearch/faiss.git synced 2026-10-11 22:50:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add docker image for CUDA 10.1. (#1477)

Browse Source 
Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1477

Test Plan: Imported from OSS

Reviewed By: wickedfoo

Differential Revision: D24492173

Pulled By: beauby

fbshipit-source-id: 5247accb2dc31bb125f9b06fb2275346b2e6465f
This commit is contained in:
Lucas Hosseini
2020-10-22 19:47:13 -07:00
committed by Facebook GitHub Bot
parent 616dc44e1e
commit 0b365fa6d8
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
conda/Dockerfile.cuda10.1
+18
View File
@@ -0,0 +1,18 @@
FROM nvidia/cuda:10.1-devel-ubuntu18.04
RUN apt-get update && apt-get install -y --allow-downgrades --allow-change-held-packages wget git libcublas-dev=10.1.0.105-1
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
RUN sh -c 'cp /usr/include/cublas*.h /usr/local/cuda/include/'
COPY ./ faiss
WORKDIR /faiss/conda
RUN CUDA_ARCHS="35;52;60;61;70;72;75" \
conda build faiss-gpu --variants '{ "cudatoolkit": "10.1" }' \
--no-anaconda-upload -c pytorch --no-test