megvii-research / CVPR2023-UniDistill

CVPR2023 (highlight) - UniDistill: A Universal Cross-Modality Knowledge Distillation Framework for 3D Object Detection in Bird's-Eye View
Apache License 2.0
101 stars 10 forks source link

[笔记]在py3.8或更高版本运行环境配置的方法 #13

Open Puiching-Memory opened 3 months ago

Puiching-Memory commented 3 months ago

注意事项:

  1. 库中带有3个预编译so文件,这导致只有py3.6可以运行,解决方法:重新编译
  2. 库中有两个文件之间只有大小写的区别,请不要直接使用git clone,会导致其中一个文件被忽略 这个文件位置

依据https://github.com/megvii-research/CVPR2023-UniDistill/issues/2

Puiching-Memory commented 3 months ago

pytorch1.10.0 python3.8 ubuntu20.04 cuda11.3 3080Ti * 1 [通过测试]

安装环境流程:

pip install mmcv-full==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html pip install mmdet==2.20.0 pip install mmseg pip install spconv-cu113==2.1.21 pip install pytorch_lightning==1.9.5

手动安装mmdet3d==0.18.0 wget https://github.com/open-mmlab/mmdetection3d/archive/refs/tags/v0.18.0.zip cd mmdet3d 别这么做##pip install -r requirements.txt pip install -v -e .

手动安装mmseg==0.30.0 pip install -v -e .

[到达此步,依据UniDistill官方文档进行安装,但是注意将requirements.txt中的pytorch及其子库注释掉,避免安装到错误的版本]

pip install cumm-cu113==0.2.9 pip install tabulate pip install pytorch-lightning==1.8.6 pip install numpy==1.19.5 pip install scipy==1.9.3 pip install torchmetrics==0.7.3

利用此存储库编译so https://github.com/Puiching-Memory/unidis_so cd unidis_so pip install -v -e . 在build文件夹中寻找so文件 将此so替换原so [然后,此存储库可以卸载]

依据https://github.com/megvii-research/CVPR2023-UniDistill/issues/2 手动下载https://github.com/megvii-research/CVPR2023-UniDistill/blob/main/unidistill/data/multisensorfusion/nuScenes_multimodal.py 放置到data/mult.../

如果使用AutoDL服务器,你可以查看以下笔记:

科学加速 source /etc/network_turbo

下载压缩zip工具 curl -L -o /usr/bin/arc http://autodl-public.ks3-cn-beijing.ksyun.com/tool/arc && chmod +x /usr/bin/arc arc compress ./1.zip ./2

运行评估

checkpoints/fusion2camera/checkpoint/f2c_submit.pth