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

Fix cuda/cudnn version detection error (#17853)

* fix cuda/cudnn version detection error, test=develop

* fix again, test=develop
上级 14a32bf0
...@@ -268,12 +268,14 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place) ...@@ -268,12 +268,14 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place)
size_t cudnn_dso_ver = dynload::cudnnGetVersion(); size_t cudnn_dso_ver = dynload::cudnnGetVersion();
LOG_FIRST_N(WARNING, 1) << "device: " << place_.device LOG_FIRST_N(WARNING, 1) << "device: " << place_.device
<< ", cuDNN Version: " << cudnn_dso_ver / 1000 << "." << ", cuDNN Version: " << cudnn_dso_ver / 1000 << "."
<< (cudnn_dso_ver % 100) / 10 << "."; << (cudnn_dso_ver % 1000) / 100 << ".";
{ {
// Check CUDA/CUDNN version compatiblity // Check CUDA/CUDNN version compatiblity
auto local_cuda_version = runtime_version_ / 100; auto local_cuda_version =
auto compile_cuda_version = CUDA_VERSION / 100; (driver_version_ / 1000) * 10 + (driver_version_ % 100) / 10;
auto compile_cuda_version =
(CUDA_VERSION / 1000) * 10 + (CUDA_VERSION % 100) / 10;
if (local_cuda_version < compile_cuda_version) { if (local_cuda_version < compile_cuda_version) {
LOG_FIRST_N(WARNING, 1) LOG_FIRST_N(WARNING, 1)
<< "WARNING: device: " << place_.device << "WARNING: device: " << place_.device
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册