Anaconda环境准备
首先你需要下载Anaconda,博主下载的是Anaconda最新版,从Anaconda官网进行下载
然后需要创建虚拟环境
conda create -n fairmot
conda activate fairmot
然后安装依赖项,这里我并没有使用原项目中建议的:pytorch==1.7.0 torvision==0.8.0 cudatoolkit==10.2
首先在终端中输入:nvidia-smi查看主机的cuda版本(如果没有下载cuda则可以先查阅一下资料如何安装,配置cuda)
之后在网站网站中找到cudatoolkit版本对应的pytorch版本然后复制指令进行下载。我下载的是torch==1.10.0 torchvision==0.11.0 cudatoolkit=11.3.1
项目代码准备
# 克隆源码
git clone https://github.com/ifzhang/FairMOT.git
# 进入项目根目录
cd {FAIRMOT_ROOT}
# 安装依赖项
pip install cython
# 之后直接运行 pip install -r requirements.txt一般会报错,需要先下载cython_bbox,可以在网上搜索压缩包,解压后到目录下执行
python setup.py build develop
# 回到{FAIRMOT_ROOT},执行
pip install -r requirements.txt
DCN_V2编译
这个是重中之重,刚开始我使用FairMOT项目中建议的pytorch版本和cudetoolkit版本编译出了很多错,也是就纠缠了好久,最后重新创建一个虚拟环境重新安装依赖项之后就可以了
# 克隆DCN_V2项目源码
git clone https://github.com/rathaROG/DCNv2_Windows.git DCNv2
cd DCNv2
# 编译源码
start make.bat
最后祝朋友们配置顺利!