Question

根据这个答案（https://groups.google.com/forum/#!msg/torch7/rmIcBYCiFG8/IC68Xzd3DgAJ），火炬似乎会自动释放不再使用的张量。而我已经有这个问题是张量指针，由lua自动释放，导致c程序中的段错误。但是，根据我的实验，似乎不能在C代码上手动调用free导致内存泄漏，程序内存不断增加。

这是我的测试代码：

#include <iostream>
#include <chrono>
#include <thread>

#include <lua.hpp>
extern "C" {
#include <TH.h>
#include <luaT.h>
}

using namespace std;

int main(int argc, char** argv)
{
    lua_State* L = luaL_newstate();

    luaL_openlibs(L);

    lua_getglobal(L, "require");
    lua_pushstring(L, "torch");
    lua_pcall(L,1,0,0); // require "torch"

    int h=224, w=224, nb_channel=3;

    int len = h * w * nb_channel;
    long stride_1 = h * w;
    long stride_2 = w;
    long stride_3 = 1;

    for (size_t i = 0 ; i < 1000000 ; ++i)
    {
        float *tensorData = (float*)malloc(sizeof(float)*len); // Should be when the storage is freed
        THFloatStorage *storage = THFloatStorage_newWithData(tensorData, len); // Memory released when tensor input is freed by lua garbadge collector
        THFloatTensor* input = THFloatTensor_newWithStorage3d(storage,
                                                              0, // Offset
                                                              nb_channel, stride_1, // Channels
                                                              h,          stride_2, // Height
                                                              w,          stride_3); // Width
        // Do some initialisation of the tensor...

        luaT_pushudata(L, (void*) input, "torch.FloatTensor"); // Send the tensor to lua
        // Do some stuff with the input... (call lua torch scripts...)
        lua_pop(L, 1);

        // If those two lines is not set, we get a memory leak
        THFloatTensor_free(input); // Will  sometimes create a segfault as the tensor seems to be deleted by the lua garbadge collector
        THFloatStorage_free(storage);

        if (i%10000 == 0)
            std::cout << i << std::endl;
        std::this_thread::sleep_for(std::chrono::microseconds(1));
    }



    return 0;
}

如何使用c和lua正确管理内存以避免在没有内存泄漏的情况下发生段错误？

Answer 1

我认为你因为没有为这行中的“tensordata”释放内存而导致内存泄漏 float tensorData =（float ）malloc（sizeof（float）* len）;

希望有所帮助

lua垃圾收集从C推出的火炬张量？

1 个答案: