找不到GPU
Created by: liudatutu
环境: docker版本:Docker version 19.03.12, build 48a66213fe nvidia-docker为官方最新版本 dock镜像文件为:hub.baidubce.com/paddlepaddle/paddle:latest-gpu-cuda10.1-cudnn7 显卡:P100/2080ti/1080ti
问题: 跑demo的时候只能用CPU,不能使用GPU,报错如下: root@265664d2942d:/app# CUDA_VISIBLE_DEVICES=0 python3 tools/infer/predict_system.py --image_dir="./doc/imgs_en/img_12.jpg" --det_model_dir="./inference/ch_det_r50_vd_db/" --rec_model_dir="./inference/ch_rec_r34_vd_crnn_enhance/" --use_space_char=True
................................
Error Message Summary:
Error: id must less than GPU count [Hint: Expected id < GetCUDADeviceCount(), but received id:0 >= GetCUDADeviceCount():0.] at (/paddle/paddle/fluid/platform/gpu_info.cc:240) 这意思就是找不到GPU; 但是我用nvidia-smi可以正常显示我GPU信息 随后我用paddle.fluid查看GPU gpu=fluid.CUDAPlace(0) 提示错误: E0806 01:12:37.937582 65 pybind.cc:1223] Cannot use GPU because there is no GPU detected on your machine.
这是赤裸裸的找不到GPU啊
请问这是官方docker镜像无效吗?内安装的cuda和cudnn不可用???? 在线求解惑!!!!!!