zyx-torch

zyx 机器学习库的 Libtorch 后端。

Libtorch 后端使用 libtorch C++ 库来执行图。

此后端需要访问 libtorch C++ 库。zyx-torch 使用 tch-rs 库作为后端，因此可以使用以下方法安装 libtorch

LibTorch 可以直接从 PyTorch 网站 下载。选择稳定版、您的操作系统、LibTorch 和您首选的计算平台。下载 cxx11 ABI。下载后，将存档提取到某个文件夹中。然后运行以下命令，并用您提取 libtorch 的文件夹路径替换 /path/to/libtorch。使用绝对路径。这将为 libtorch 设置必要的环境变量。请注意，在您重新启动计算机时，环境变量将被重置，因此您需要重新导出它们，或者将它们放入自动运行的脚本中。

export LIBTORCH=/path/to/libtorch
export LD_LIBRARY_PATH=${LIBTORCH}/lib:$LD_LIBRARY_PATH

请阅读 PyTorch 文档了解所有先决条件，您可能需要安装 C++ 编译器或 GPU 驱动程序。

如果您已安装 PyTorch，则上述环境变量将与其冲突。在这种情况下，您可能更喜欢使用 PyTorch 的 libtorch

export LIBTORCH_USE_PYTORCH=1

如果上述选项都不适用于您，zyx-torch 可以通过启用下载 libtorch 功能自动安装 libtorch

有关 README 和源代码，请访问 github。

有关更多详细信息，请参阅 书籍。

Cargo 功能

• std - 启用 zyx-core/std
• download-libtorch - 安装 libtorch