fix(cpu): cudnn is now on by default, auto switch it to off in case o…

…f cpu_only
jolibrain · Sep 17, 2021 · 3770baf · 3770baf
1 parent 74487dc
commit 3770baf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -258,6 +258,7 @@ if (USE_CPU_ONLY)
   set(USE_CAFFE2_CPU_ONLY ON)
   set(USE_TORCH_CPU_ONLY ON)
   set(USE_FAISS_CPU_ONLY ON)
+  set(USE_CUDNN OFF)
 else()
   cmake_minimum_required(VERSION 3.14)
   include(cmake/Cuda.cmake) # cuda + cudnn