fix_cudnn_compatible_check

6 years ago · 9793a0b6a6
parent ccb322d6a5
commit 9793a0b6a6
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/platform/device_context.cc
+++ b/paddle/fluid/platform/device_context.cc
@ -292,7 +292,7 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place)
    if (dynload::HasCUDNN()) {
      auto local_cudnn_version = cudnn_dso_ver / 100;
      auto compile_cudnn_version = CUDNN_VERSION / 100;
-      if (local_cuda_version < compile_cuda_version) {
+      if (local_cudnn_version < compile_cudnn_version) {
        LOG_FIRST_N(WARNING, 1)
            << "WARNING: device: " << place_.device
            << ". The installed Paddle is compiled with CUDNN "