Test providing CUDA path.

facebookresearch · Jan 7, 2021 · 82fbc47 · 82fbc47
1 parent db48ec9
commit 82fbc47
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -107,6 +107,11 @@ jobs:
       image: windows-server-2019-nvidia:stable
     steps:
       - checkout
+      - run:
+          command: |
+            cd "C:/Program Files (x86)/Microsoft Visual Studio/2019/Community/MSBuild/Microsoft/VC/"
+            cp v160/BuildCustomizations/CUDA* v150/BuildCustomizations/
+            cp v160/BuildCustomizations/Nvda.* v150/BuildCustomizations/
       - run:
           name: Build/test
           command: |

diff --git a/conda/faiss-gpu/build-lib.bat b/conda/faiss-gpu/build-lib.bat
@@ -4,7 +4,7 @@
 :: LICENSE file in the root directory of this source tree.
 
 :: Build libfaiss.so.
-cmake -B _build ^
+CUDA_PATH="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/" cmake -B _build ^
 -T v141 ^
 -A x64 ^
 -G "Visual Studio 16 2019" ^
@@ -15,6 +15,8 @@ cmake -B _build ^
 -DFAISS_ENABLE_PYTHON=OFF ^
 -DBLA_VENDOR=Intel10_64_dyn ^
 .
+cat _build/CMakeFiles/CMakeOutput.log
+cat _build/CMakeFiles/CMakeError.log
 if %errorlevel% neq 0 exit /b %errorlevel%
 
 cmake --build _build --config Release -j %CPU_COUNT%