NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Facundo Galán 2019-03-27 21:41:46 -03:00 committed by Davis E. King
parent ae406bf4c1
commit 64ad6b2591
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dlib/cuda/gpu_data.cpp
View File

@ -83,7 +83,7 @@ namespace dlib
{
#if !defined CUDA_VERSION
#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
// reason makes training freeze in some cases.
// (see https://github.com/davisking/dlib/issues/1513)