We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 23f874d commit 04a2387Copy full SHA for 04a2387
dlib/cuda/gpu_data.cpp
@@ -11,7 +11,7 @@
11
#include <iostream>
12
#include "cuda_utils.h"
13
#include <cstring>
14
-
+#include <cuda.h>
15
16
namespace dlib
17
{
@@ -81,7 +81,9 @@ namespace dlib
81
82
void synchronize_stream(cudaStream_t stream)
83
84
-#if CUDA_VERSION >= 9020 && CUDA_VERSION <= 10000
+#if !defined CUDA_VERSION
85
+#error CUDA_VERSION not defined
86
+#elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10000
87
// This should be pretty much the same as cudaStreamSynchronize, which for some
88
// reason makes training freeze in some cases.
89
// (see https://github.com/davisking/dlib/issues/1513)
0 commit comments