[package]
edition = "2024"
rust-version = "1.85"
name = "tract-cuda"
version = "0.22.1"
authors = [
"Louis Chouraki <louis.chouraki@sonos.com>",
"Mathieu Poumeyrol <kali@zoy.org>",
]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Tiny, no-nonsense, self contained, TensorFlow and ONNX inference"
readme = false
keywords = [
"TensorFlow",
"NeuralNetworks",
"CUDA",
]
categories = ["science"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/snipsco/tract"
resolver = "2"
[badges.maintenance]
status = "actively-developed"
[lib]
name = "tract_cuda"
path = "src/lib.rs"
[dependencies.anyhow]
version = "1.0.43"
[dependencies.cudarc]
version = "0.16.4"
features = [
"dynamic-loading",
"cuda-12060",
"f16",
]
[dependencies.derive-new]
version = "0.5.9"
[dependencies.downcast-rs]
version = "1.2.0"
[dependencies.log]
version = "0.4.14"
[dependencies.num-traits]
version = "0.2.14"
[dependencies.tract-core]
version = "=0.22.1"
[dependencies.tract-gpu]
version = "=0.22.1"
[dependencies.tract-transformers]
version = "=0.22.1"
[dev-dependencies.proptest]
version = "1.0.0"
[dev-dependencies.rand]
version = "0.8.4"
features = ["small_rng"]