From c289aa97823e980f5376d1159f804da2aae75590 Mon Sep 17 00:00:00 2001 From: Sanjoy Das Date: Mon, 2 Nov 2020 01:26:12 -0800 Subject: [PATCH] PR #43950: Print the version of the cuDNN library found at runtime Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/43950 This is especially useful if we build against X.Y.z and run with X.Y.w where w != z. Copybara import of the project: -- e5183817a249adc6998e7dd29c70974bf2b61221 by Sanjoy Das : Print the version of the cuDNN library found at runtime This is especially useful if we build against X.Y.z and run with X.Y.w where w != z. COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/43950 from sanjoy:print-cudnn-version e5183817a249adc6998e7dd29c70974bf2b61221 PiperOrigin-RevId: 340187371 Change-Id: If7251624ed2599ced7ac50f8f5e501bdf46e6b59 --- tensorflow/stream_executor/cuda/cuda_dnn.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/stream_executor/cuda/cuda_dnn.cc b/tensorflow/stream_executor/cuda/cuda_dnn.cc index 8fa70d500a0..907b1c74996 100644 --- a/tensorflow/stream_executor/cuda/cuda_dnn.cc +++ b/tensorflow/stream_executor/cuda/cuda_dnn.cc @@ -340,6 +340,8 @@ port::Status CudnnSupport::Init() { #endif cudnn_.reset(new CudnnAccess(cudnn_handle)); + + LOG(INFO) << "Loaded cuDNN version " << cudnnGetVersion(); return port::Status::OK(); } -- GitLab