Getting std::bad_alloc when loading the python tradined model in c++

1

Here is the python code :

image
image1081×382 66.6 KB

cpp code :

#include <iostream>
#include <memory>
#include <torch/script.h>

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/cpu/libtorch-shared-with-deps-2.5.1%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)

error while running cpp :

terminate called after throwing an instance of ‘std::bad_alloc’
what(): std::bad_alloc
Aborted (core dumped)