记录一下trackformer的安装过程

RedMery

已于 2024-02-15 23:23:09 修改

阅读量813

点赞数

文章标签： trackformer pytorch

于 2023-07-23 19:14:52 首次发布

本文链接：https://blog.csdn.net/u010948546/article/details/131882768

版权

1、创建python环境，并激活

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

在这里插入图片描述

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

在这里插入图片描述

# 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

在这里插入图片描述

pip install -r requirements.txt

在这里插入图片描述

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

最后直接运行的时候，提示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

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

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

在这里插入图片描述

关注