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