未验证 提交 d88c3e12 编写于 作者: P pangyoki 提交者: GitHub

Expose paddle.version.show API and add doc for it (#36800)

* add doc for show() in paddle.version

* fix format

* print cuda and cudnn in show API
上级 c93331c5
...@@ -100,9 +100,50 @@ istaged = %(istaged)s ...@@ -100,9 +100,50 @@ istaged = %(istaged)s
commit = '%(commit)s' commit = '%(commit)s'
with_mkl = '%(with_mkl)s' with_mkl = '%(with_mkl)s'
__all__ = ['cuda', 'cudnn'] __all__ = ['cuda', 'cudnn', 'show']
def show(): def show():
"""Get the version of paddle if `paddle` package if tagged. Otherwise, output the corresponding commit id.
Returns:
If paddle package is not tagged, the commit-id of paddle will be output.
Otherwise, the following information will be output.
full_version: version of paddle
major: the major version of paddle
minor: the minor version of paddle
patch: the patch level version of paddle
rc: whether it's rc version
cuda: the cuda version of package. It will return `False` if CPU version paddle package is installed
cudnn: the cudnn version of package. It will return `False` if CPU version paddle package is installed
Examples:
.. code-block:: python
import paddle
# Case 1: paddle is tagged with 2.2.0
paddle.version.show()
# full_version: 2.2.0
# major: 2
# minor: 2
# patch: 0
# rc: 0
# cuda: '10.2'
# cudnn: '7.6.5'
# Case 2: paddle is not tagged
paddle.version.show()
# commit: cfa357e984bfd2ffa16820e354020529df434f7d
# cuda: '10.2'
# cudnn: '7.6.5'
"""
if istaged: if istaged:
print('full_version:', full_version) print('full_version:', full_version)
print('major:', major) print('major:', major)
...@@ -111,6 +152,8 @@ def show(): ...@@ -111,6 +152,8 @@ def show():
print('rc:', rc) print('rc:', rc)
else: else:
print('commit:', commit) print('commit:', commit)
print('cuda:', cuda_version)
print('cudnn:', cudnn_version)
def mkl(): def mkl():
return with_mkl return with_mkl
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册