tensorrt 0.1.0

Rust wrapper for NVIDIA TensorRT
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

[package]
name = "tensorrt"
version = "0.1.0"
authors = ["Ming Yang <ymviv@qq.com>"]
description = "Rust wrapper for NVIDIA TensorRT"
repository = "https://github.com/vivym/tensorrt-rs"
readme = "README.md"
keywords = ["tensorrt", "nvidia", "cuda", "ffi"]
license = "MIT/Apache-2.0"
edition = "2021"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
cuda-rs = "0.1"
tensorrt-rs-sys = "0.1"
thiserror = "1"

[dev-dependencies]
clap = { version = "4", features = ["derive"] }
tch = "0.14.0"