mirror of https://github.com/AlexeyAB/darknet.git
Compile fix
This commit is contained in:
parent
7e7964388b
commit
dcfeea30f1
|
@ -481,7 +481,7 @@ void show_cuda_cudnn_info()
|
||||||
CHECK_CUDA(cudaRuntimeGetVersion(&cuda_version));
|
CHECK_CUDA(cudaRuntimeGetVersion(&cuda_version));
|
||||||
CHECK_CUDA(cudaDriverGetVersion(&cuda_driver_version));
|
CHECK_CUDA(cudaDriverGetVersion(&cuda_driver_version));
|
||||||
fprintf(stderr, " CUDA-version: %d (%d)", cuda_version, cuda_driver_version);
|
fprintf(stderr, " CUDA-version: %d (%d)", cuda_version, cuda_driver_version);
|
||||||
if(cuda_version < cuda_driver_version) printf(stderr, "\n Warning: CUDA-version is lower than Driver-version! \n");
|
if(cuda_version < cuda_driver_version) fprintf(stderr, "\n Warning: CUDA-version is lower than Driver-version! \n");
|
||||||
#ifdef CUDNN
|
#ifdef CUDNN
|
||||||
fprintf(stderr, ", cuDNN: %d.%d.%d", CUDNN_MAJOR, CUDNN_MINOR, CUDNN_PATCHLEVEL);
|
fprintf(stderr, ", cuDNN: %d.%d.%d", CUDNN_MAJOR, CUDNN_MINOR, CUDNN_PATCHLEVEL);
|
||||||
#endif // CUDNN
|
#endif // CUDNN
|
||||||
|
|
Loading…
Reference in New Issue