提交 e030741d 编写于 作者: T tensor-tang

fix gpu fraction

上级 746a62eb
...@@ -72,6 +72,8 @@ void MainThreads(int num_threads, bool use_gpu) { ...@@ -72,6 +72,8 @@ void MainThreads(int num_threads, bool use_gpu) {
NativeConfig config; NativeConfig config;
config.model_dir = FLAGS_dirname + "word2vec.inference.model"; config.model_dir = FLAGS_dirname + "word2vec.inference.model";
config.use_gpu = use_gpu; config.use_gpu = use_gpu;
config.fraction_of_gpu_memory = 0.15;
config.device = 0;
auto main_predictor = auto main_predictor =
CreatePaddlePredictor<NativeConfig, PaddleEngineKind::kNative>(config); CreatePaddlePredictor<NativeConfig, PaddleEngineKind::kNative>(config);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册