本文记录BEVFormer复现中需要注意的一些地方

torch相关安装

在官方教程中,给了推荐的torch的版本。经过实验,pytorch不要超过1.10,因为它取消了一个特性。
[THC/THC.h: No such file or directory报错解决][1]

1
2
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
# Recommended torch>=1.9

mmcv-full的安装

如果按照官方的做法,往往会卡住(科学上网可能会比较顺利),下面提供其它方法。
我们通过mmcv-full官方推荐的方法进行下载。
[Building wheel for mmcv-full (setup.py) … error解决办法][2]

1
2
pip install -U openmim
mim install mmcv-full==1.4.0

安装mmdet and mmseg

正常安装即可

mmdet3d源码安装

由于是用的服务器镜像,torch版本和python版本对应不上。需要自行安装torch。

镜像的选择

[我应该选择哪个 Docker 镜像?][3]
Alpine:极小的镜像大小,快速的构建和部署时间,较低的资源消耗。可能存在软件包的新颖性和兼容性问题,较少的库支持。
Bookworm:提供最新的安全更新和软件包。
Bullseye:提供长期的支持和维护。

[1]https://blog.csdn.net/qq_40349484/article/details/134528674
[2]https://blog.csdn.net/weixin_46412818/article/details/126006081
[3]https://cloud.tencent.com/developer/article/2457031