Is it possible to extract features using 11GB GPU?

Question

Is it possible to extract features using 11GB GPU?

Opened this issue 3 years ago · 1 comments

Now, the process reports F0625 08:25:59.166216 119 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory on my 11GB 2080ti. But that's all we have.

Answer 1 · 2021-06-25T08:28:09.000Z

When I load the model into GPU, it costs only 1.8G, but the feature extraction requires way larger than that.