关于CUDA 10.2和CUDNN 10.2 ,v 7.6.5.32 下在:
一、安装python
版本python 39即可
更新pip
python -m pip install --upgrade pip
更新protobuf
pip3 install protobuf==3.19.0
二、安装CUDA 10.2和CUDNN
我的机器是win10,m4000的n卡。当前版本CUDA11.6,为了使用,从nvidia 官网下载
CUDA 10.2 和 Cudnn 7.6.5 for windows
CUDA10.2下载需要注册,之前装的时候没有记录网址,比较难找,只记录了CUDNN
Download cuDNN v7.6.5 (November 18th, 2019), for CUDA 10.2
三、安装paddlepaddle
python -m pip install paddlepaddle-gpu==2.3.0 -i https://mirror.baidu.com/pypi/simple
# 在您的Python解释器中确认PaddlePaddle安装成功
>>> import paddle
>>> paddle.utils.run_check()
四、安装vs等相关控件
简单粗暴从微软官网下载 并安装 14.0的库
五、安装paddledetection
git clone https://github.com/PaddlePaddle/PaddleDetection.git
# 安装其他依赖 从国内镜像下载
cd PaddleDetection
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
# 编译安装paddledet
python setup.py install
六、解决cocoapi问题
pip install git+https://gitee.com/linuxonly801/philferriere-cocoapi.git#subdirectory=PythonAPI
七、识别
python deploy/python/infer.py --model_dir=C:\Users\yzx\Desktop\paddle\PaddleDetection\fire_model --video_file=img/test1.mp4 --device=gpu
百度随便弄两张图片来识别