Bevformer环境配置

1. 创建环境并激活环境

conda create -n open-mmlab python=3.8 -y
conda activate open-mmlab

2. 安装pytorch、Cuda、CuDnn

# 建议使用清华源
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch

3. 安装gcc

conda install -c omgarcia gcc-6 # gcc-6.2

4. 安装 mmcv-full

安装mmcv指导文档
pip安装总会出错，那么久可以换成mim安装。

pip install openmim
# 不要用清华源 才用原本的默认源
mim install mmcv-full==1.4.0

5. 安装 mmdet3d

min install mmdet3d==0.17.1
# 如果不起作用需要在github下载代码进行编译
pip install -v -e.

6. 安装必要的依赖

pip install einops fvcore seaborn iopath==0.1.9 timm==0.6.13  typing-extensions==4.5.0 pylint ipython==8.12  numpy==1.19.5 matplotlib==3.5.2 numba==0.48.0 pandas==1.4.4 scikit-image==0.19.3 setuptools==59.5.0

conda install git

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'

训练和测试

./tools/dist_train.sh ./projects/configs/bevformer/bevformer_tiny.py 1