(Paste the commands you ran that produced the error.)
What have you tried to solve it?
add a method to delete index memory which created by method ai.djl.pytorch.jni.PyTorchLibrary#torchIndexInit
Environment Info
Please run the command ./gradlew debugEnv from the root directory of DJL (if necessary, clone DJL first). It will output information about your system, environment, and installation that can help us debug your issue. Paste the output of the command below:
Description
PtNDArray.get(NDArray index)和PtNDArray.set(NDIndex index, NDArray value) cause memory leak
Expected Behavior
(what's the expected behavior?)
Error Message
memory leak
How to Reproduce?
try (final NDManager ndManager = NDManager.newBaseManager()) { while (true) { try (NDManager manager = ndManager.newSubManager()) { NDArray a = manager.create(new int[1024 1024 1]); final NDArray eq = a.eq(1); a.get(eq); } } }
Steps to reproduce
(Paste the commands you ran that produced the error.)
What have you tried to solve it?
Environment Info
Please run the command
./gradlew debugEnv
from the root directory of DJL (if necessary, clone DJL first). It will output information about your system, environment, and installation that can help us debug your issue. Paste the output of the command below: