文件名称:PyTorch 的 C++ api 的 Rust 绑定。
文件大小:854KB
文件格式:ZIP
更新时间:2024-07-18 11:46:29
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} /