Make pytorch available in CircleCI. (#1486)

Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/1486

Reviewed By: wickedfoo

Differential Revision: D24493486

Pulled By: beauby

fbshipit-source-id: 00156213061503ff593b2e9ede062850b23527a9
This commit is contained in:
Lucas Hosseini
2020-10-22 21:10:32 -07:00
committed by Facebook GitHub Bot
parent 7891094da6
commit bab6db84e0
2 changed files with 4 additions and 3 deletions
+2 -2
View File
@@ -7,5 +7,5 @@ sudo apt-get install -y python3-dev python3-pip swig libmkl-dev
# Install recent CMake.
RUN wget -nv -O - https://github.com/Kitware/CMake/releases/download/v3.17.1/cmake-3.17.1-Linux-x86_64.tar.gz | sudo tar xzf - --strip-components=1 -C /usr
# Install numpy/scipy for python tests.
RUN pip3 install numpy scipy
# Install numpy/scipy/pytorch for python tests.
RUN pip3 install numpy scipy torch
+2 -1
View File
@@ -79,7 +79,7 @@ jobs:
- run:
name: Test python extension
command: |
pip3 install pytest
pip3 install pytest torch
export PYTHONPATH="$(ls -d ./build/faiss/python/build/lib*/)"
pytest --junitxml=test-results/pytest/results.xml ./tests
- store_test_results:
@@ -95,6 +95,7 @@ jobs:
name: Build/test
command: |
conda install conda-build
conda install -c pytorch pytorch
cd conda
conda build faiss --python 3.7