Test providing CUDA path.

facebookresearch · Jan 11, 2021 · 2671e14 · 2671e14
1 parent db48ec9
commit 2671e14
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
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
@@ -8,13 +8,16 @@ cmake -B _build ^
 -T v141 ^
 -A x64 ^
 -G "Visual Studio 16 2019" ^
+-DCMAKE_VS_PLATFORM_TOOLSET_CUDA="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1" ^
 -DBUILD_SHARED_LIBS=ON ^
 -DBUILD_TESTING=OFF ^
 -DFAISS_ENABLE_GPU=ON ^
 -DCMAKE_CUDA_ARCHITECTURES="%CUDA_ARCHS%" ^
 -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%