安装python库
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
pip install mmengine
pip install mmcv-2.0.0rc4-cp38-cp38-manylinux1_x86_64.whl
pip install 'mmdet>=3.0.0,<3.1.0'
cd mmdetection3d
pip install -v -e .
python库检查
python -c 'import torch;print(torch.__version__);print(torch.version.cuda)'
1.9.1+cu111
11.1
python -c 'import mmengine;print(mmengine.__version__);'
0.10.6
python -c 'import mmcv;print(mmcv.__version__);'
2.0.0rc4
python -c 'import mmdet;print(mmdet.__version__);'
3.3.0
python -c 'import numpy;print(numpy.__version__);'
1.24.4
推理验证
下载模型权重