提交 c289aa97 编写于 作者: S Sanjoy Das 提交者: TensorFlower Gardener

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 <sanjoy@playingwithpointers.com>:

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
上级 942e7121
......@@ -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();
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册