tch-rs:PyTorch的C ++ API的Rust绑定

时间:2024-02-25 07:25:18
【文件属性】:

文件名称:tch-rs:PyTorch的C ++ API的Rust绑定

文件大小:643KB

文件格式:ZIP

更新时间:2024-02-25 07:25:18

rust machine-learning deep-learning neural-network pytorch

频道 PyTorch的C ++ API的Rust绑定。 tch板条箱的目标是在C ++ PyTorch api(又名libtorch)周围提供一些薄包装纸。 它旨在保持与原始C ++ API的距离。 然后可以在此基础上开发更多惯用的防锈绑定。 该可以在docs.rs上找到。 libtorch之上的C api的代码生成部分来自 。 入门 这个箱子需要在版本V1.7.0的C ++ PyTorch库(libtorch)可用您的系统上。 您可以手动安装它,并通过LIBTORCH环境变量让构建脚本知道它。 如果未设置,则构建脚本将尝试下载并提取libtorch的预构建二进制版本。 Libtorch手动


网友评论