NVIDIA/TensorRT.git:

【升级通知】Gitee 将于 7 月 11 日晚 23 点进行全站升级维护

代码拉取完成,页面将自动刷新

richgiteeai

Watch

不关注关注所有动态仅关注版本发行动态关注但不提醒动态

1 Star 0 Fork 0

kotte/TensorRT

代码 Issues 0 Pull Requests 0 Wiki 统计流水线

服务

加入 Gitee

与超过 1400万开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)

免费加入

已有帐号? 立即登录

文件

新建文件新建 Diagram 文件

main

分支 (18)

标签 (31)

管理

main

dependabot/pip/tensorflow-gpu-2.5.3

dependabot/pip/samples/python/network_api_pytorch_mnist/pillow-9.0.0

dependabot/pip/samples/python/introductory_parser_samples/pillow-9.0.0

dependabot/pip/samples/python/engine_refit_mnist/pillow-9.0.0

dependabot/pip/samples/python/int8_caffe_mnist/pillow-9.0.0

dependabot/pip/samples/python/end_to_end_tensorflow_mnist/pillow-9.0.0

dependabot/pip/samples/sampleSSD/pillow-9.0.0

dependabot/pip/samples/python/uff_custom_plugin/pillow-9.0.0

dependabot/pip/pillow-9.0.0

release/8.2

master

release/8.0

release/7.2

release/7.1

release/7.0

release/6.0

release/5.1

22.02

22.01

8.2.1

8.2.0-EA

21.10

21.09

21.08

21.07

8.0.1

21.06

21.05

21.04

21.03

21.02

20.12

20.11

20.10

7.2.1

20.09

20.08

克隆/下载

HTTPS SSH SVN SVN+SSH 下载ZIP

提示

下载代码请复制以下命令到终端执行

为确保你提交的代码身份被 Gitee 正确识别,请执行以下命令完成配置

git config --global user.name userName 
git config --global user.email userEmail

初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置

1 生成 RSA 密钥

2 获取 RSA 公钥内容,并配置到 SSH公钥中

在 Gitee 上使用 SVN,请访问使用指南

使用 HTTPS 协议时,命令行会出现如下账号密码验证步骤。基于安全考虑,Gitee 建议配置并使用私人令牌替代登录密码进行克隆、推送等操作

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # 私人令牌

新建文件新建 Diagram 文件

分支 18

标签 31

贡献代码

同步代码

创建 Pull Request

了解更多

对比差异通过 Pull Request 同步

同步更新到分支

通过 Pull Request 同步

将会在向当前分支创建一个 Pull
Request,合入后将完成同步

Yinghai Lu Adding a iter binding to Dims 34a1675

257 次提交

File empty ...

docstrings

include

packaging

src

CMakeLists.txt

README.md

build.sh

requirements.txt

README

# TensorRT Python Bindings

## Installation

### Download pybind11

Create a directory for external sources and download pybind11 into it.
```bash
export EXT_PATH=~/external

mkdir -p $EXT_PATH && cd $EXT_PATH
git clone https://github.com/pybind/pybind11.git
```

### Download Python headers

#### Add Main Headers

1. Get the source code from the official [python sources](https://www.python.org/downloads/source/)
2. Copy the contents of the `Include/` directory into `$EXT_PATH/pythonX.Y/include/` directory.

#### Add PyConfig.h

1. Download the deb package for the desired platform from [here](https://packages.debian.org/search?searchon=contents&keywords=pyconfig.h&mode=path&suite=unstable&arch=any).
 Typical plaforms include `x86_64` (`amd64`), `aarch64` (`arm64`), and `ppc64le` (`ppc64el`)
2. Unpack the debian with `ar x <libpython...>.deb`
3. Unpack the contained `data.tar.xz` with `tar -xvf`
4. Copy the `./usr/include/<platform>/` directory into the `$$EXT_PATH/pythonX.Y/include/` directory here.
 It should only contain a single file - `pyconfig.h`

### Build Python bindings

Use `build.sh` to generate the installable wheels for intended python version and target architecture.

Example: for python 3.8 `x86_64` wheel,
```bash
cd $TRT_OSSPATH/python
PYTHON_MAJOR_VERSION=3 PYTHON_MINOR_VERSION=8 TARGET_ARCHITECTURE=x86_64 ./build.sh
```

### Install the python wheel

```bash
python3 -m pip install build/dist/tensorrt-*.whl
```