Merge pull request #103120 from danieldk/libtorch-bin-cmake-fix-lib-paths
libtorch-bin: fix CMake library paths for split outputs
This commit is contained in:
commit
da115b4c56
@ -45,6 +45,14 @@ in stdenv.mkDerivation {
|
||||
|
||||
# We do not care about Java support...
|
||||
rm -f $out/lib/lib*jni* 2> /dev/null || true
|
||||
|
||||
# Fix up library paths for split outputs
|
||||
substituteInPlace $dev/share/cmake/Torch/TorchConfig.cmake \
|
||||
--replace \''${TORCH_INSTALL_PREFIX}/lib "$out/lib" \
|
||||
|
||||
substituteInPlace \
|
||||
$dev/share/cmake/Caffe2/Caffe2Targets-release.cmake \
|
||||
--replace \''${_IMPORT_PREFIX}/lib "$out/lib" \
|
||||
'';
|
||||
|
||||
postFixup = let
|
||||
|
@ -1,3 +1,4 @@
|
||||
cmake_minimum_required(VERSION 3.0)
|
||||
find_package(Torch REQUIRED)
|
||||
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
|
||||
add_executable(test test.cpp)
|
||||
|
Loading…
Reference in New Issue
Block a user