库版本控制

我们提供版本号宏来识别正在使用的 LibTorch 版本。示例用法

#include <torch/torch.h>
#include <iostream>

int main() {
  std::cout << "PyTorch version from parts: "
    << TORCH_VERSION_MAJOR << "."
    << TORCH_VERSION_MINOR << "."
    << TORCH_VERSION_PATCH << std::endl;
  std::cout << "PyTorch version: " << TORCH_VERSION << std::endl;
}

这将输出类似以下内容

PyTorch version from parts: 1.8.0
PyTorch version: 1.8.0

注意

这些宏仅在 PyTorch >= 1.8.0 中可用。