Std::bad_alloc при загрузке модели PyTorch на C++ с использованием LibTorch ⇐ Python

[Anonymous]

Я пытаюсь загрузить отследил модель PyTorch на C++ с помощью LibTorch, но обнаружил ошибку std::bad_alloc. Я привел минимальный пример, чтобы продемонстрировать проблему.
Проблема:
При запуске моего приложения C++, загружающего модель PyTorch, вылетает из-за:
terminate called after throwing an instance of 'std::bad_alloc'
what(): std::bad_alloc
Aborted (core dumped)

Код cpp:
#include
#include
#include

int main()
{
// Load the traced model
torch::jit::script::Module module;

// Load the saved model - update path to your .pt file
module = torch::jit::load("net.pt");

return 0;
}

CMakeLists.txt
cmake_minimum_required(VERSION 3.18)
project(torch_example)

include(FetchContent)

set(LIBTORCH_CPU_URL "https://download.pytorch.org/libtorch/c ... %2Bcpu.zip")

FetchContent_Declare(
Torch
URL ${LIBTORCH_CPU_URL }
DOWNLOAD_EXTRACT_TIMESTAMP TRUE
)
FetchContent_MakeAvailable(Torch)

add_executable(torch_example main.cpp)
target_link_libraries(torch_example PRIVATE "${TORCH_LIBRARIES}")
target_include_directories(torch_example PRIVATE "${torch_SOURCE_DIR}/include")
set_target_properties(torch_example PROPERTIES CXX_STANDARD 23)


Подробнее здесь: https://stackoverflow.com/questions/791 ... g-libtorch