未验证 提交 eb9e4601 编写于 作者: P pangyoki 提交者: GitHub

add cudnn error (#47666)

上级 4a4f3f80
......@@ -108,6 +108,20 @@ void InitGpuProperties(Place place,
<< "Please recompile or reinstall Paddle with compatible CUDA "
"version.";
}
auto local_cudnn_version = cudnn_dso_ver / 1000;
auto compile_cudnn_version = CUDNN_VERSION / 1000;
PADDLE_ENFORCE_EQ(
compile_cudnn_version,
local_cudnn_version,
phi::errors::InvalidArgument(
"The installed Paddle is compiled with CUDNN "
"%d, but CUDNN version in your machine is %d. "
"which will cause serious incompatible bug. "
"Please recompile or reinstall Paddle with compatible CUDNN "
"version.",
compile_cudnn_version,
local_cudnn_version));
#endif
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册