记录一下trackformer的安装过程

该文详细描述了如何创建Python环境并激活,然后在Trackformer项目中安装必要的依赖,包括PyTorch、requirements.txt中的库。当遇到MultiScaleDeformableAttention安装错误时,由于CUDA和GCC版本不兼容,文章提供了降级GCC到10版并建立与CUDA链接的解决方案,最后安装成功后,使用ffmpeg制作图片并开始对象检测。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

项目地址

1、创建python环境,并激活

conda create -n TF python=3.8 -y
conda activate TF

在这里插入图片描述

2、进入trackformer中,并且运行setup.py

cd Desktop/MOT/trackformer/
python setup.py install

在这里插入图片描述

3、下载pytorch

pytorch官方安装法

# CUDA 11.1
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述

4、安装 requirements.txt

pip install -r requirements.txt
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

在这里插入图片描述

5、缺什么库 就装什么库

pip install pycocotools
pip install sacred==0.8.4
pip install pyyaml

6、安装MultiScaleDeformableAttention

最后直接运行的时候,提示MultiScaleDeformableAttention
在这里插入图片描述
因此需要安装MultiScaleDeformableAttention

python src/trackformer/models/ops/setup.py build --build-base=src/trackformer/models/ops/ install

在这里插入图片描述
在这里插入图片描述
然而有的时候会报错,报错内容如下图所是,从中可以看出,报错的愿意是cuda和gcc的版本不兼容。可以看出,gcc的版本不应该超过10.
参考cuda error gcc(unsupported GNU version! gcc versions later than 5 are not supported)在这里插入图片描述输出gcc的版本

gcc -v

在这里插入图片描述安装gcc10

sudo apt-get install gcc-10
sudo apt-get install g++-10

将gcc10与cuda建立软连接

sudo ln -s /usr/bin/gcc-10 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++10 /usr/local/cuda/bin/g++

安装MultiScaleDeformableAttention

python src/trackformer/models/ops/setup.py build --build-base=src/trackformer/models/ops/ install

7 make picture

sudo apt install ffmpeg
ffmpeg -i data/snakeboard/snakeboard.mp4 -vf fps=30 data/snakeboard/%06d.png

8、开始检测

python src/track.py with dataset_name=DEMO data_root_dir=data/snakeboard output_dir=data/snakeboard write_images=pretty

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

python解决报错ImportError: Bad git executable.
https://blog.csdn.net/zzpl139/article/details/128228698
running build running build_ext building 'MultiScaleDeformableAttention' extension Traceback (most recent call last): File "src/trackformer/models/ops/setup.py", line 56, in <module> setup( File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/__init__.py", line 117, in setup return distutils.core.setup(**attrs) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/core.py", line 183, in setup return run_commands(dist) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/core.py", line 199, in run_commands dist.run_commands() File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 954, in run_commands self.run_command(cmd) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/dist.py", line 950, in run_command super().run_command(command) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 973, in run_command cmd_obj.run() File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/command/build.py", line 135, in run self.run_command(cmd_name) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/cmd.py", line 316, in run_command self.distribution.run_command(command) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/dist.py", line 950, in run_command super().run_command(command) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/_distutils/dist.py", line 973, in run_command cmd_obj.run() File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/setuptools/command/build_ext.py", line 98, in run _build_ext.run(self) File "/home/zzz/anaconda3/envs/trackformer/lib/python3.8/site-packages/Cython/Distutils/old_buil
06-29
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值