`CUDA_VISIBLE_DEVICES` 没有设置 | 没有属性 `gpu_predictor`
Created by: DrRyanHuang
在AIstudio
运行demo“PaddleHub人脸检测示例”:
https://aistudio.baidu.com/aistudio/projectdetail/215962
将module.face_detection
中的use_gpu
设置为True
报错,说环境变量没有正确设置:
RuntimeError: Attempt to use GPU for prediction, but environment variable CUDA_VISIBLE_DEVICES was not set correctly.
我尝试设置环境变量:
%set_env CUDA_VISIBLE_DEVICES=0
再次执行:module.face_detection
再次报错:
AttributeError: 'FaceDetector640' object has no attribute 'gpu_predictor'
这是环境的问题还是paddlehub的问题?
麻烦帮忙看一下