代码的下载地址
其中包含paddle各种行为识别的训练-推理-导出代码。
2.打架行为的推理测试代码(可将长视频按照每7帧抽一帧,累计8帧,给出一个分类结果)
GitHub - PaddlePaddle/PaddleDetection at develop
服务器配置:
1.去paddle官网根据自己系统和cuda的版本选择适合的paddle-gpu版本进行安装:
python -m pip install paddlepaddle-gpu==2.3.1 -i https://mirror.baidu.com/pypi/simple
2.安装nccl
根据cuda
的版本去选择对应版本的nccl,可以去nvidia的官网下载
https://developer.nvidia.com/nccl/nccl-legacy-downloads
按照官方的安装教程安装即可,一定要确认好cuda的版本。
将nccl添加到环境变量中
首先,找到你nccl的安装目录,你问我怎么找?当然是终端输入命令:whereis nccl 了,我的是在/usr/include/nccl.h中
然后,终端输入vim ~/.bashrc进入该文件,添加如下内容到文件中(添加到最低行):
#设置cuda库的目录
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64
#将nccl添加到LD_LIBRARY_PATH中
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/include/nccl.h
最后,保存好了,咱还要更新,让配置文件生效啊,终端输入命令:source ~/.bashrc 再通过echo $LD_LIBRARY_PATH命令查看环境变量设置存在,若存在则添加成功。