提交 9793a0b6 编写于 作者: S sneaxiy

fix_cudnn_compatible_check

上级 ccb322d6
...@@ -292,7 +292,7 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place) ...@@ -292,7 +292,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.
先完成此消息的编辑!
想要评论请 注册