diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
index ce854fe527f6c1529aeea34715b7b085cb9562bf..53ca49e64a9d9ba8d5cafe6d8a387d8fbe3db674 100644
--- a/cpp/CMakeLists.txt
+++ b/cpp/CMakeLists.txt
@@ -71,6 +71,7 @@ set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -O0 -g")
 message("CUDA_TOOLKIT_ROOT_DIR=${CUDA_TOOLKIT_ROOT_DIR}")
 message("CUDA_NVCC_FLAGS=${CUDA_NVCC_FLAGS}")
 
+set(GPU_VERSION "ON")
 add_definitions("-DGPU_VERSION")
 
 set(CMAKE_CXX_STANDARD 14)
diff --git a/cpp/src/CMakeLists.txt b/cpp/src/CMakeLists.txt
index a76986de3481cb081e3f967720c636d0029e11fd..d81ccc5da2dd53643824548fbe01f2c0becb5344 100644
--- a/cpp/src/CMakeLists.txt
+++ b/cpp/src/CMakeLists.txt
@@ -62,7 +62,7 @@ set(third_party_libs
         sqlite
         thrift
         yaml-cpp
-        gpufaiss
+        libgpufaiss.a
         faiss
         lapack
         openblas