1.官网安装指南:
Windows下的Conda安装-使用文档-PaddlePaddle深度学习平台
这边有环境要求,以及paddlepaddle和paddledetection对应版本。
一定要对应好。。不然会出问题。(第一次cudnn是7.4.5的)装完报错。
所以先检查一下自己的cuda版本和cudnn版本
cuda版本,打开cmd,输入
nvcc -V
以上需要安装好CUDA和cudnn对应的版本,详情可参考Pytorch安装教程:
Windows10 下搭建Yolov5运行环境:Anaconda3、CUDA与cudnn、Pytorch等_NCUTer的博客-CSDN博客_yolo运行环境
1.paddledetection是paddlepaddle下的一个子模块。所以必须要先安装paddlepaddle才能运行paddledetection
2.打开cmd,先利用conda创建虚拟环境
conda create -n paddle_env python=3.8
根据自己的cuda版本去官网选择安装命令。我这边是cuda10.2
paddlepaddle官网:https://www.paddlepaddle.org.cn/
对于国内用户无法连接到Anaconda官方源的可以按照以下命令添加清华源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/conda config --set show_channel_urls yes
4.激活虚拟环境进行安装
conda install paddlepaddle-gpu==2.3.1 cudatoolkit=10.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/
验证是否安装成功
python -c "import paddle; print(paddle.__version__)"
安装成功显示
2.2.3
5.安装paddledetection
仓库:https://github.com/PaddlePaddle/PaddleDetection
将其下载到本地并解压
cd到有requirments.txt的目录下,安装其他依赖项
输入
- pip install -r requirments.txt
- python setup.py install
方法二:pip安装
注意: pip安装方式只支持Python3
pip install paddledet==2.4.0 -i https://mirror.baidu.com/pypi/simple
6.安装pycocotools,yolo训练要要到coco的评价指标(根据情况选择安装)
window下输入
pip install pycocotools-windows
百度 通过conda安装可以成功
conda install -c conda-forge lap
7.测试paddledetection
python ppdet/modeling/tests/test_architectures.py