mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Minor fix

Browse Source
This commit is contained in:
AlexeyAB
2019-03-28 00:11:16 +03:00
parent 7b1ef39607
commit f5fd086407
6 changed files with 30 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/dark_cuda.c
View File

@ -25,6 +25,10 @@ int gpu_index = 0;
#endif // USE_CMAKE_LIBS
#endif // CUDNN
#if defined(CUDNN_HALF) && !defined(CUDNN)
#error "If you set CUDNN_HALF=1 then you must set CUDNN=1"
#endif
void cuda_set_device(int n)
{
gpu_index = n;