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
ありがとうございました!