Compile fix

This commit is contained in:
AlexeyAB 2019-12-29 15:12:22 +03:00
parent 7e7964388b
commit dcfeea30f1
1 changed files with 1 additions and 1 deletions

View File

@ -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