prepare_environment.md
September 10, 2024 ยท View on GitHub
Prepare Environment
- Linux
- python 3.8
- Pytorch 1.10.0+
- CUDA 11.1+
1. Create a conda virtual environment
conda create -n simpb python=3.8 -y
conda activate simpb
2. Install Pytorch
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
3. Install other packages
pip install --upgrade pip
pip install -r requirement.txt
4. Compile the deformable_aggregation CUDA op
cd projects/mmdet3d_plugin/ops
python setup.py develop
cd ../../../