Skip to content

Latest commit

 

History

History
35 lines (24 loc) · 961 Bytes

File metadata and controls

35 lines (24 loc) · 961 Bytes

Install the SegFormer++ without MMSegmentation/MMPose

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.0

If it is still not working, make sure to use a pip virtual environment (python -m venv "name of the environment"), not a conda environment.