Skip to content

GitLab

Commit 64ad6b25 authored by Facundo Galán's avatar Facundo Galán Committed by Davis E. King
Browse files

Change CUDA version check for cudaSync from 10000 to 10010 (#1704)

parent ae406bf4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
dlib/cuda/gpu_data.cpp
View file @ 64ad6b25
...@@ -83,7 +83,7 @@ namespace dlib ...@@ -83,7 +83,7 @@ namespace dlib
{ {
#if !defined CUDA_VERSION #if !defined CUDA_VERSION
#error CUDA_VERSION not defined #error CUDA_VERSION not defined
#elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10000 #elif CUDA_VERSION >= 9020 && CUDA_VERSION <= 10010
// This should be pretty much the same as cudaStreamSynchronize, which for some // This should be pretty much the same as cudaStreamSynchronize, which for some
// reason makes training freeze in some cases. // reason makes training freeze in some cases.
// (see https://github.com/davisking/dlib/issues/1513) // (see https://github.com/davisking/dlib/issues/1513)
......
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