【发布时间】:2021-08-11 14:06:33
【问题描述】:
我正在尝试编译一个使用 LibTorch for android 的程序。
但是由script/build_android.sh 在pytorch repo 中构建的库只包含torch.jit 模块。我需要torch.nn 模块。
是否可以访问完整的 C++ LibTorch API?如果是这样,我该如何构建它?
【问题讨论】:
标签: android c++ mobile pytorch libtorch
我正在尝试编译一个使用 LibTorch for android 的程序。
但是由script/build_android.sh 在pytorch repo 中构建的库只包含torch.jit 模块。我需要torch.nn 模块。
是否可以访问完整的 C++ LibTorch API?如果是这样,我该如何构建它?
【问题讨论】:
标签: android c++ mobile pytorch libtorch
似乎通过在根 CMake 文件中将 NO_API 设置为 OFF 并将 BUILD_MOBILE_AUTOGRAD 设置为 ON,您可以构建 api 并因此构建 torch.nn。
【讨论】: