get_cudnn_version

paddle.get_cudnn_version ( ) [源代码]

此函数返回cudnn的版本。 返回值是int,它表示cudnn版本。 例如,如果返回7600,则表示cudnn的版本为7.6。

返回:返回一个整数,表示cudnn的版本。

代码示例

  1. import paddle
  2. device = paddle.get_cudnn_version()