0

pytorch c++ ライブラリを CMake とリンクするには、本当に必要find_package(Torch REQUIRED)で使用するだけです

-DCMAKE_PREFIX_PATH=/absolute/path/to/libtorch

ソースはこちらからダウンロードできます: https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip CMake のチュートリアルはこちら: https://pytorch.org/cppdocs/ installation.html

私の中には次のものがありますWORKSPACE

new_local_repository(
    name = "torch",
    build_file_content = all_content,
    path = "third_party/libtorch",
)

そして、これは私の中でBUILD

cmake_external(
    name = "torch",
    cache_entries = {
        "CMAKE_PREFIX_PATH": "/home/jackshi/projects/third_party/libtorch",
    },
    lib_source = "@torch//:all",
)

リンクしようとすると、<dir> does not appear to contain CMakeList.txt これは本当ですが、でリンクするとfind_package(Torch REQUIRED)、トップレベルCMakeList.txtは必要ありません。CMake は を探しますTorchConfig.cmake。bazel はトップレベルなしでこのパッケージのターゲットを作成できますCMakeList.txtか? 探しているはずTorchConfig.cmakeですよね?

CMAKE_PREFIX_PATHまた、ファイルがリモートで取得された場合、どのように機能しますか?ディレクトリhttp_archiveを使用しますか?~/.cache/bazel

ありがとうございました!

4

1 に答える 1