Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add docs for tensorrt and cuda_archs version #6859

Open
wants to merge 2 commits into
base: develop
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions docs/api/paddle/version/Overview_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,10 @@ paddle.version 目录下包含的 API 返回 paddle 安装包相关配置的版
:widths: 10, 30

" :ref:`cuda <cn_api_paddle_version_cuda>` ", "获取 paddle wheel 包编译时使用的 CUDA 版本"
" :ref:`cuda_archs <cn_api_paddle_version_cuda_archs>` ", "获取 paddle wheel 包编译时的 CUDA 架构"
" :ref:`cudnn <cn_api_paddle_version_cudnn>` ", "获取 paddle wheel 包编译时使用的 cuDNN 版本"
" :ref:`nccl <cn_api_paddle_version_nccl>` ", "获取 paddle wheel 包编译时使用的 nccl 版本"
" :ref:`tensorrt <cn_api_paddle_version_tensorrt>` ", "获取 paddle wheel 包编译时使用的 TensorRt 版本"
" :ref:`xpu <cn_api_paddle_version_xpu>` ", "获取 paddle wheel 包编译时使用的 XPU 版本"
" :ref:`xpu_xccl <cn_api_paddle_version_xpu_xccl>` ", "获取 paddle wheel 包编译时使用的 XPU xccl 版本"
" :ref:`show <cn_api_paddle_version_show>` ", "打印 paddle 版本、CUDA 版本、cuDNN 版本等信息"
19 changes: 19 additions & 0 deletions docs/api/paddle/version/cuda_archs_cn.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
.. _cn_api_paddle_version_cuda_archs:

cuda_archs
-------------------------------

.. py:function:: paddle.version.cuda_archs()

获取 paddle 安装包编译时的 CUDA 架构。


返回
:::::::::

返回 paddle 安装包编译时的 CUDA 架构列表。

代码示例:
::::::::::

COPY-FROM: paddle.version.cuda_archs
2 changes: 2 additions & 0 deletions docs/api/paddle/version/show_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ show
- rc - 是否是 rc 版本。
- cuda - 若 paddle wheel 包为 GPU 版本,则返回 paddle wheel 包编译时使用的 CUDA 的版本信息;若 paddle wheel 包为 CPU 版本,则返回 ``False`` 。
- cudnn - 若 paddle wheel 包为 GPU 版本,则返回 paddle wheel 包编译时使用的 cuDNN 的版本信息;若 paddle wheel 包为 CPU 版本,则返回 ``False`` 。
- tensorrt - 返回 paddle 安装包编译时使用的 TensorRT 版本号,若无安装 TensorRT, 则返回 None 。
- cuda_archs - 返回 paddle 安装包编译时的 CUDA 架构列表。

代码示例
::::::::::
Expand Down
19 changes: 19 additions & 0 deletions docs/api/paddle/version/tensorrt_cn.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
.. _cn_api_paddle_version_tensorrt:

tensorrt
-------------------------------

.. py:function:: paddle.version.tensorrt()

获取 paddle 安装包编译时使用的 TensorRT 版本号。


返回
:::::::::

返回 paddle 安装包编译时使用的 TensorRT 版本号,若无安装 TensorRT, 则返回 None 。

代码示例:
::::::::::

COPY-FROM: paddle.version.tensorrt