Commit 1e623983 authored by Davis King's avatar Davis King

Fixed cmake script so nvcc doesn't freak out.

parent 9e2f26fc
...@@ -412,7 +412,10 @@ if (NOT TARGET dlib) ...@@ -412,7 +412,10 @@ if (NOT TARGET dlib)
if (CUDA_FOUND) if (CUDA_FOUND)
set(CUDA_HOST_COMPILATION_CPP ON) set(CUDA_HOST_COMPILATION_CPP ON)
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_30;-std=c++11") # Note that we add __STRICT_ANSI__ to avoid freaking out nvcc with gcc specific
# magic in the standard C++ header files (since nvcc uses gcc headers on
# linux).
list(APPEND CUDA_NVCC_FLAGS "-arch=sm_30;-std=c++11;-D__STRICT_ANSI__")
message(STATUS "Looking for cuDNN install...") message(STATUS "Looking for cuDNN install...")
# Look for cudnn, we will look in the same place as other CUDA # Look for cudnn, we will look in the same place as other CUDA
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
#include <stdlib.h> #include <stdlib.h>
#include <stdio.h> #include <stdio.h>
//#include "cuda_dlib.h" #include "cuda_dlib.h"
#define CHECK(call) \ #define CHECK(call) \
{ \ { \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment