Step 0. Prerequisites
- Pytorch: 2.3 (CUDA 12.1) (older versions should also work fine)
Step 1. Install MMCV using MIM.
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"Step 2. Install Segformer++
cd model
pip install .Step 3. Run the SegFormer++
Troubleshooting There might be installation troubles with openmim, mmengine, and mmcv for new python versions. Thus, Step 1 might not work correctly. In this case, try the following alternative for step 1:
pip install torch torchvision numpy
pip install wheel==0.45.1
pip install mmcv==2.2.0If it is still not working, make sure to use a pip virtual environment (python -m venv "name of the environment"), not a conda environment.