在C++应用程序中使用Tensorflow :如何释放图形处理器内存

#include <tensorflow/core/public/session.h>
#include <tensorflow/core/platform/env.h>

tensorflow::Session* session;
tensorflow::SessionOptions session_options;
tensorflow::Status status = tensorflow::NewSession(session_options, &session);

tensorflow::GraphDef graph_def;
status = tensorflow::ReadBinaryProto(tensorflow::Env::Default(), "path/to/your/model.pb", &graph_def);
status = session->Create(graph_def);

std::vector<tensorflow::Tensor> outputs;
status = session->Run({}, {}, {"your_output_tensor_name"}, &outputs);

session->Close();