PaddleClas PP-ShiTu运行示例时，提示显存不足，不能使用GPU

您好，我在使用PP-Shitu体验时，遇到了GPU识别报错的问题。我的使用环境如下： PaddleClas版本：2.3 PaddlePaddle版本:2.1.3 CUDA:10.2 cudnn:7.6.5 系统：Ubuntu18.04 Python:3.8.10

当我运行https://github.com/PaddlePaddle/PaddleClas/blob/release/2.3/docs/zh_CN/quick_start/quick_start_recognition.md 该文档进行操作时报错如下： python python/predict_system.py -c configs/inference_general.yaml 2021-11-12 10:03:13 INFO:

2021-11-12 10:03:13 INFO: DetPostProcess : 2021-11-12 10:03:13 INFO: DetPreProcess : 2021-11-12 10:03:13 INFO: transform_ops : 2021-11-12 10:03:13 INFO: DetResize : 2021-11-12 10:03:13 INFO: interp : 2 2021-11-12 10:03:13 INFO: keep_ratio : False 2021-11-12 10:03:13 INFO: target_size : [640, 640] 2021-11-12 10:03:13 INFO: DetNormalizeImage : 2021-11-12 10:03:13 INFO: is_scale : True 2021-11-12 10:03:13 INFO: mean : [0.485, 0.456, 0.406] 2021-11-12 10:03:13 INFO: std : [0.229, 0.224, 0.225] 2021-11-12 10:03:13 INFO: DetPermute : 2021-11-12 10:03:13 INFO: Global : 2021-11-12 10:03:13 INFO: batch_size : 1 2021-11-12 10:03:13 INFO: cpu_num_threads : 10 2021-11-12 10:03:13 INFO: det_inference_model_dir : ./models/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer 2021-11-12 10:03:13 INFO: enable_benchmark : True 2021-11-12 10:03:13 INFO: enable_mkldnn : True 2021-11-12 10:03:13 INFO: enable_profile : False 2021-11-12 10:03:13 INFO: gpu_mem : 8000 2021-11-12 10:03:13 INFO: image_shape : [3, 640, 640] 2021-11-12 10:03:13 INFO: infer_imgs : ./drink_dataset_v1.0/test_images/nongfu_spring.jpeg 2021-11-12 10:03:13 INFO: ir_optim : True 2021-11-12 10:03:13 INFO: labe_list : ['foreground'] 2021-11-12 10:03:13 INFO: max_det_results : 5 2021-11-12 10:03:13 INFO: rec_inference_model_dir : ./models/general_PPLCNet_x2_5_lite_v1.0_infer 2021-11-12 10:03:13 INFO: rec_nms_thresold : 0.05 2021-11-12 10:03:13 INFO: threshold : 0.2 2021-11-12 10:03:13 INFO: use_fp16 : False 2021-11-12 10:03:13 INFO: use_gpu : True 2021-11-12 10:03:13 INFO: use_tensorrt : False 2021-11-12 10:03:13 INFO: IndexProcess : 2021-11-12 10:03:13 INFO: index_dir : ./drink_dataset_v1.0/index/ 2021-11-12 10:03:13 INFO: return_k : 5 2021-11-12 10:03:13 INFO: score_thres : 0.5 2021-11-12 10:03:13 INFO: RecPostProcess : None 2021-11-12 10:03:13 INFO: RecPreProcess : 2021-11-12 10:03:13 INFO: transform_ops : 2021-11-12 10:03:13 INFO: ResizeImage : 2021-11-12 10:03:13 INFO: size : 224 2021-11-12 10:03:13 INFO: NormalizeImage : 2021-11-12 10:03:13 INFO: mean : [0.485, 0.456, 0.406] 2021-11-12 10:03:13 INFO: order : 2021-11-12 10:03:13 INFO: scale : 0.00392157 2021-11-12 10:03:13 INFO: std : [0.229, 0.224, 0.225] 2021-11-12 10:03:13 INFO: ToCHWImage : None E1112 10:03:13.824231 6743 analysis_predictor.cc:679] Allocate too much memory for the GPU memory pool, assigned 8000 MB E1112 10:03:13.824254 6743 analysis_predictor.cc:682] Try to shink the value by setting AnalysisConfig::EnableGpu(...) Inference: 22.044897079467773 ms per batch image [{'bbox': [244, 49, 509, 964], 'rec_docs': '农夫山泉-饮用天然水', 'rec_scores': 0.75856656}]

结果确实是正确的，但log中提示我的显卡显存不足。请问运行pp-shitu需要的显存是多大呢?或者是我哪里出了问题，谢谢

Nov 12 '21 02:11 gaooolianggg

我又试了一下paddleDetection和paddleCls的分类示例，都能够正常运行。而且我在windows系统下也会报同样的错误，那么是显存确实不够吗？我的显卡是2070，显存8G

Nov 12 '21 02:11 gaooolianggg

正确运行峰值在1.2G左右, V100上测试；试试先把后台耗费显存的任务先杀掉，然后单独运行，监控下显存消耗

Nov 12 '21 02:11 Intsigstephon

正确运行峰值在1.2G左右, V100上测试；试试先把后台耗费显存的任务先杀掉，然后单独运行，监控下显存消耗. 我确定后台没有其他占用显存的任务。而且从log上看，程序要分配的显存达到了8000M

Nov 12 '21 03:11 gaooolianggg

找到解决办法了。其实就在代码中运行的那个config文件中有一个设置显存大小的参数，把那个参数设置小一些就可以了。希望其他人遇到这个问题可以解决

Nov 12 '21 05:11 gaooolianggg

PaddleClas PaddleClas copied to clipboard

PP-ShiTu运行示例时，提示显存不足，不能使用GPU

PaddleClas
PaddleClas copied to clipboard