diff --git a/tensorflow/stream_executor/cuda/cuda_dnn.cc b/tensorflow/stream_executor/cuda/cuda_dnn.cc index 8fa70d500a0db3713a89951061be96505d6f51a8..907b1c74996dc3a8cdda023e310c634ede7c3e42 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(); }