PyTorch 的 C++ api 的 Rust 绑定。

【文件属性】：
文件名称：PyTorch 的 C++ api 的 Rust 绑定。
文件大小：854KB
文件格式：ZIP
更新时间：2021-06-28 17:59:49
rust machine-learning deep-learning neural-network pytorch tch-rsPyTorch 的 C++ api 的 Rust 绑定。tch crate 的目标是围绕 C++ PyTorch api（又名 libtorch）提供一些薄包装。它旨在尽可能接近原始 C++ api。然后可以在此基础上开发更多惯用的 Rust 绑定。文档可以在 docs.rs 上找到。libtorch 之上的 C api 的代码生成部分来自ocaml-torch 。入门这个箱子需要在版本V1.7.0的C ++ PyTorch库（libtorch）可用您的系统上。您可以手动安装它并通过LIBTORCH环境变量让构建脚本知道它。如果未设置，构建脚本将尝试下载并提取 libtorch 的预构建二进制版本。Libtorch 手动安装从PyTorch 网站下载部分获取libtorch并解压缩 zip 文件的内容。对于 Linux 用户，将以下内容添加到.bashrc或等效文件中，其中/path/to/libtorch是解压缩文件时创建的目录的路径。export LIBTORCH=/path/to/libtorchexport LD_LIBRARY_PATH=${LIBTORCH} /