correctly set var

mlverse · Aug 5, 2023 · 251a0b8 · 251a0b8
1 parent ea5a42a
commit 251a0b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lantern/CMakeLists.txt b/src/lantern/CMakeLists.txt
@@ -18,7 +18,7 @@ if (DEFINED ENV{CUDA} AND NOT "$ENV{CUDA}" STREQUAL "")
     set(CUDA_VERSION "$ENV{CUDA}")
     message(STATUS "CUDA VERSION: $ENV{CUDA} | ${CUDA_VERSION} | ${CUDA_VERSION_NUMBER}")
 
-    set(USE_CUDNN 1)
+    set(CAFFE2_USE_CUDNN 1)
 endif()