未验证 提交 c562be20 编写于 作者: Z Zeng Jinle 提交者: GitHub

Merge pull request #15193 from sneaxiy/fix_cudnn_compatible_check

Fix cudnn compatible check
...@@ -285,7 +285,7 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place) ...@@ -285,7 +285,7 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place)
if (dynload::HasCUDNN()) { if (dynload::HasCUDNN()) {
auto local_cudnn_version = cudnn_dso_ver / 100; auto local_cudnn_version = cudnn_dso_ver / 100;
auto compile_cudnn_version = CUDNN_VERSION / 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) LOG_FIRST_N(WARNING, 1)
<< "WARNING: device: " << place_.device << "WARNING: device: " << place_.device
<< ". The installed Paddle is compiled with CUDNN " << ". The installed Paddle is compiled with CUDNN "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册