MMDetection3D v1.3.0安装教程
- 1. 系统环境
- 2. 安装
- 2.1 基本环境安装
- 2.2 调整具体版本
- 2.3 验证
- 2.4 安装MinkowskiEngine和TorchSparse
- 3. 最终环境配置
根据 v1.3.0版本官方手册测试后的安装配置，亲测可行。
1. 系统环境
| 项目 | 版本 | 日期 |
|---|---|---|
| Ubuntu | 18.04.06 LTS | - |
| 显卡 | RTX 2070 | - |
| 显卡驱动 | 525.105.17 | - |
2. 安装
2.1 基本环境安装
验证过的安装配置
conda create --name openmmlab130 python=3.8 -y
conda activate openmmlab130
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install -U openmim
mim install mmengine
mim install 'mmcv>=2.0.0rc4'
mim install 'mmdet>=3.0.0rc5, <3.3.0'
git clone https://github.com/open-mmlab/mmdetection3d.git -b v1.3.0
cd mmdetection3d130
pip install -v -e .
pip install -r requirements/optional.txt
pip install cumm-cu118 && pip install spconv-cu1182.2 调整具体版本
pip install numba==0.55.2
pip install pandas==1.4.4
pip install protobuf==3.20.1
pip install open3d
2.3 验证
mim download mmdet3d --config pointpillars_hv_secfpn_8xb6-160e_kitti-3d-car --dest .
python demo/pcd_demo.py demo/data/kitti/000008.bin pointpillars_hv_secfpn_8xb6-160e_kitti-3d-car.py hv_pointpillars_secfpn_6x8_160e_kitti-3d-car_20220331_134606-d42d15ed.pth --show
2.4 安装MinkowskiEngine和TorchSparse
注意，将其中的路径换为自己虚拟环境对应的链接，并且在pip安装过程中关闭其他程序，因为编译过程需要消耗大量的内存和CPU。
根据自己的环境下载并在系统安装CUDA，用于这两个库的编译
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
export CUDA_HOME=/usr/local/cuda-11.8
conda install openblas-devel -c anaconda
export CPLUS_INCLUDE_PATH=CPLUS_INCLUDE_PATH:/home/bit2/anaconda3/envs/openmmlab130/include
pip install -U git+https://github.com/NVIDIA/MinkowskiEngine -v --no-deps --config-settings="--blas_include_dirs=/home/bit2/anaconda3/envs/openmmlab130/include" --config-settings="--blas=openblas"
sudo apt-get install libsparsehash-dev
pip install --upgrade git+https://github.com/mit-han-lab/torchsparse.git@v1.4.03. 最终环境配置
# Name Version Build Channel
_libgcc_mutex 0.1 main defaults
_openmp_mutex 5.1 1_gnu defaults
absl-py 0.15.0 pypi_0 pypi
addict 2.4.0 pypi_0 pypi
aliyun-python-sdk-core 2.14.0 pypi_0 pypi
aliyun-python-sdk-kms 2.16.2 pypi_0 pypi
appdirs 1.4.4 pypi_0 pypi
asttokens 2.4.1 pypi_0 pypi
astunparse 1.6.3 pypi_0 pypi
attrs 23.2.0 pypi_0 pypi
backcall 0.2.0 pypi_0 pypi
black 20.8b1 pypi_0 pypi
blas 1.0 mkl defaults
blinker 1.7.0 pypi_0 pypi
brotli-python 1.0.9 py38h6a678d5_7 defaults
bzip2 1.0.8 h7b6447c_0 defaults
ca-certificates 2023.08.22 h06a4308_0 anaconda
cachetools 5.3.3 pypi_0 pypi
ccimport 0.4.2 pypi_0 pypi
certifi 2023.11.17 py38h06a4308_0 anaconda
cffi 1.16.0 pypi_0 pypi
charset-normalizer 2.0.4 pyhd3eb1b0_0 defaults
clang 5.0 pypi_0 pypi
click 8.1.7 pypi_0 pypi
colorama 0.4.6 pypi_0 pypi
comm 0.2.1 pypi_0 pypi
configargparse 1.7 pypi_0 pypi
contourpy 1.1.1 pypi_0 pypi
crcmod 1.7 pypi_0 pypi
cryptography 42.0.5 pypi_0 pypi
cuda-cudart 11.8.89 0 nvidia
cuda-cupti 11.8.87 0 nvidia
cuda-libraries 11.8.0 0 nvidia
cuda-nvrtc 11.8.89 0 nvidia
cuda-nvtx 11.8.86 0 nvidia
cuda-runtime 11.8.0 0 nvidia
cumm-cu118 0.4.11 pypi_0 pypi
cycler 0.12.1 pypi_0 pypi
dash 2.15.0 pypi_0 pypi
dash-core-components 2.0.0 pypi_0 pypi
dash-html-components 2.0.0 pypi_0 pypi
dash-table 5.0.0 pypi_0 pypi
decorator 5.1.1 pypi_0 pypi
descartes 1.1.0 pypi_0 pypi
exceptiongroup 1.2.0 pypi_0 pypi
executing 2.0.1 pypi_0 pypi
fastjsonschema 2.19.1 pypi_0 pypi
ffmpeg 4.3 hf484d3e_0 pytorch
filelock 3.13.1 py38h06a4308_0 defaults
fire 0.5.0 pypi_0 pypi
flake8 7.0.0 pypi_0 pypi
flask 3.0.2 pypi_0 pypi
flatbuffers 1.12 pypi_0 pypi
fonttools 4.49.0 pypi_0 pypi
freetype 2.12.1 h4a9f257_0 defaults
gast 0.4.0 pypi_0 pypi
gmp 6.2.1 h295c915_3 defaults
gmpy2 2.1.2 py38heeb90bb_0 defaults
gnutls 3.6.15 he1e5248_0 defaults
google-auth 2.28.1 pypi_0 pypi
google-auth-oauthlib 1.0.0 pypi_0 pypi
google-pasta 0.2.0 pypi_0 pypi
grpcio 1.62.0 pypi_0 pypi
h5py 3.1.0 pypi_0 pypi
idna 3.4 py38h06a4308_0 defaults
imageio 2.34.0 pypi_0 pypi
importlib-metadata 7.0.1 pypi_0 pypi
importlib-resources 6.1.2 pypi_0 pypi
iniconfig 2.0.0 pypi_0 pypi
intel-openmp 2023.1.0 hdb19cb5_46306 defaults
ipython 8.12.3 pypi_0 pypi
ipywidgets 8.1.2 pypi_0 pypi
itsdangerous 2.1.2 pypi_0 pypi
jedi 0.19.1 pypi_0 pypi
jinja2 3.1.3 py38h06a4308_0 defaults
jmespath 0.10.0 pypi_0 pypi
joblib 1.3.2 pypi_0 pypi
jpeg 9e h5eee18b_1 defaults
jsonschema 4.21.1 pypi_0 pypi
jsonschema-specifications 2023.12.1 pypi_0 pypi
jupyter-core 5.7.1 pypi_0 pypi
jupyterlab-widgets 3.0.10 pypi_0 pypi
keras 2.15.0 pypi_0 pypi
keras-preprocessing 1.1.2 pypi_0 pypi
kiwisolver 1.4.5 pypi_0 pypi
lame 3.100 h7b6447c_0 defaults
lark 1.1.9 pypi_0 pypi
lazy-loader 0.3 pypi_0 pypi
lcms2 2.12 h3be6417_0 defaults
ld_impl_linux-64 2.38 h1181459_1 defaults
lerc 3.0 h295c915_0 defaults
libcublas 11.11.3.6 0 nvidia
libcufft 10.9.0.58 0 nvidia
libcufile 1.8.1.2 0 nvidia
libcurand 10.3.4.107 0 nvidia
libcusolver 11.4.1.48 0 nvidia
libcusparse 11.7.5.86 0 nvidia
libdeflate 1.17 h5eee18b_1 defaults
libffi 3.4.4 h6a678d5_0 defaults
libgcc-ng 11.2.0 h1234567_1 defaults
libgfortran-ng 8.2.0 hdf63c60_1 anaconda
libgomp 11.2.0 h1234567_1 defaults
libiconv 1.16 h7f8727e_2 defaults
libidn2 2.3.4 h5eee18b_0 defaults
libnpp 11.8.0.86 0 nvidia
libnvjpeg 11.9.0.86 0 nvidia
libopenblas 0.3.2 h9ac9557_1 anaconda
libpng 1.6.39 h5eee18b_0 defaults
libstdcxx-ng 11.2.0 h1234567_1 defaults
libtasn1 4.19.0 h5eee18b_0 defaults
libtiff 4.5.1 h6a678d5_0 defaults
libunistring 0.9.10 h27cfd23_0 defaults
libwebp-base 1.3.2 h5eee18b_0 defaults
llvmlite 0.38.1 pypi_0 pypi
lyft-dataset-sdk 0.0.8 pypi_0 pypi
lz4-c 1.9.4 h6a678d5_0 defaults
markdown 3.5.2 pypi_0 pypi
markdown-it-py 3.0.0 pypi_0 pypi
markupsafe 2.1.3 py38h5eee18b_0 defaults
matplotlib 3.5.3 pypi_0 pypi
matplotlib-inline 0.1.6 pypi_0 pypi
mccabe 0.7.0 pypi_0 pypi
mdurl 0.1.2 pypi_0 pypi
minkowskiengine 0.5.4 pypi_0 pypi
mkl 2023.1.0 h213fc3f_46344 defaults
mkl-service 2.4.0 py38h5eee18b_1 defaults
mkl_fft 1.3.8 py38h5eee18b_0 defaults
mkl_random 1.2.4 py38hdb19cb5_0 defaults
mmcv 2.1.0 pypi_0 pypi
mmdet 3.2.0 pypi_0 pypi
mmdet3d 1.3.0 dev_0 <develop>
mmengine 0.10.3 pypi_0 pypi
model-index 0.1.11 pypi_0 pypi
mpc 1.1.0 h10f8cd9_1 defaults
mpfr 4.0.2 hb69a4c5_1 defaults
mpmath 1.3.0 py38h06a4308_0 defaults
mypy-extensions 1.0.0 pypi_0 pypi
nbformat 5.9.2 pypi_0 pypi
ncurses 6.4 h6a678d5_0 defaults
nest-asyncio 1.6.0 pypi_0 pypi
nettle 3.7.3 hbbd107a_1 defaults
networkx 3.1 py38h06a4308_0 defaults
ninja 1.11.1.1 pypi_0 pypi
numba 0.55.2 pypi_0 pypi
numpy 1.19.5 pypi_0 pypi
nuscenes-devkit 1.1.11 pypi_0 pypi
oauthlib 3.2.2 pypi_0 pypi
open3d 0.18.0 pypi_0 pypi
openblas-devel 0.3.2 0 anaconda
opencv-python 4.9.0.80 pypi_0 pypi
opendatalab 0.0.10 pypi_0 pypi
openh264 2.1.1 h4ff587b_0 defaults
openjpeg 2.4.0 h3ad879b_0 defaults
openmim 0.3.9 pypi_0 pypi
openssl 3.0.12 h7f8727e_0 anaconda
openxlab 0.0.34 pypi_0 pypi
opt-einsum 3.3.0 pypi_0 pypi
ordered-set 4.1.0 pypi_0 pypi
oss2 2.17.0 pypi_0 pypi
packaging 23.2 pypi_0 pypi
pandas 1.4.4 pypi_0 pypi
parso 0.8.3 pypi_0 pypi
pathspec 0.12.1 pypi_0 pypi
pccm 0.4.11 pypi_0 pypi
pexpect 4.9.0 pypi_0 pypi
pickleshare 0.7.5 pypi_0 pypi
pillow 10.2.0 py38h5eee18b_0 defaults
pip 23.3.1 py38h06a4308_0 defaults
pkgutil-resolve-name 1.3.10 pypi_0 pypi
platformdirs 4.2.0 pypi_0 pypi
plotly 5.19.0 pypi_0 pypi
pluggy 1.4.0 pypi_0 pypi
plyfile 1.0.3 pypi_0 pypi
portalocker 2.8.2 pypi_0 pypi
prompt-toolkit 3.0.43 pypi_0 pypi
protobuf 3.20.1 pypi_0 pypi
ptyprocess 0.7.0 pypi_0 pypi
pure-eval 0.2.2 pypi_0 pypi
pyasn1 0.5.1 pypi_0 pypi
pyasn1-modules 0.3.0 pypi_0 pypi
pybind11 2.11.1 pypi_0 pypi
pycocotools 2.0.7 pypi_0 pypi
pycodestyle 2.11.1 pypi_0 pypi
pycparser 2.21 pypi_0 pypi
pycryptodome 3.20.0 pypi_0 pypi
pyflakes 3.2.0 pypi_0 pypi
pygments 2.17.2 pypi_0 pypi
pyparsing 3.1.1 pypi_0 pypi
pyquaternion 0.9.9 pypi_0 pypi
pysocks 1.7.1 py38h06a4308_0 defaults
pytest 8.0.2 pypi_0 pypi
python 3.8.18 h955ad1f_0 defaults
python-dateutil 2.8.2 pypi_0 pypi
pytorch 2.0.1 py3.8_cuda11.8_cudnn8.7.0_0 pytorch
pytorch-cuda 11.8 h7e8668a_5 pytorch
pytorch-mutex 1.0 cuda pytorch
pytz 2023.4 pypi_0 pypi
pywavelets 1.4.1 pypi_0 pypi
pyyaml 6.0.1 pypi_0 pypi
readline 8.2 h5eee18b_0 defaults
referencing 0.33.0 pypi_0 pypi
regex 2023.12.25 pypi_0 pypi
requests 2.28.2 pypi_0 pypi
requests-oauthlib 1.3.1 pypi_0 pypi
retrying 1.3.4 pypi_0 pypi
rich 13.4.2 pypi_0 pypi
rpds-py 0.18.0 pypi_0 pypi
rsa 4.9 pypi_0 pypi
scikit-image 0.19.3 pypi_0 pypi
scikit-learn 1.3.2 pypi_0 pypi
scipy 1.10.1 pypi_0 pypi
setuptools 60.2.0 pypi_0 pypi
six 1.15.0 pypi_0 pypi
spconv-cu118 2.3.6 pypi_0 pypi
sqlite 3.41.2 h5eee18b_0 defaults
stack-data 0.6.3 pypi_0 pypi
sympy 1.12 py38h06a4308_0 defaults
tabulate 0.9.0 pypi_0 pypi
tbb 2021.8.0 hdb19cb5_0 defaults
tenacity 8.2.3 pypi_0 pypi
tensorboard 2.14.0 pypi_0 pypi
tensorboard-data-server 0.7.2 pypi_0 pypi
tensorflow 2.6.0 pypi_0 pypi
tensorflow-estimator 2.15.0 pypi_0 pypi
termcolor 1.1.0 pypi_0 pypi
terminaltables 3.1.10 pypi_0 pypi
threadpoolctl 3.3.0 pypi_0 pypi
tifffile 2023.7.10 pypi_0 pypi
tk 8.6.12 h1ccaba5_0 defaults
toml 0.10.2 pypi_0 pypi
tomli 2.0.1 pypi_0 pypi
torchaudio 2.0.2 py38_cu118 pytorch
torchsparse 1.4.0 pypi_0 pypi
torchtriton 2.0.0 py38 pytorch
torchvision 0.15.2 py38_cu118 pytorch
tqdm 4.65.2 pypi_0 pypi
traitlets 5.14.1 pypi_0 pypi
trimesh 4.1.6 pypi_0 pypi
typed-ast 1.5.5 pypi_0 pypi
typing-extensions 4.10.0 pypi_0 pypi
tzdata 2024.1 pypi_0 pypi
urllib3 1.26.18 pypi_0 pypi
waymo-open-dataset-tf-2-6-0 1.4.9 pypi_0 pypi
wcwidth 0.2.13 pypi_0 pypi
werkzeug 3.0.1 pypi_0 pypi
wheel 0.41.2 py38h06a4308_0 defaults
widgetsnbextension 4.0.10 pypi_0 pypi
wrapt 1.12.1 pypi_0 pypi
xz 5.4.6 h5eee18b_0 defaults
yapf 0.40.2 pypi_0 pypi
zipp 3.17.0 pypi_0 pypi
zlib 1.2.13 h5eee18b_0 defaults
zstd 1.5.5 hc292b87_0 defaults
