| # Installation | |
| ## Requirements | |
| - Linux with Python ≥ <font color='red'>3.10</font>. | |
| - PyTorch ≥ <font color='red'>2.0</font> and [torchvision](https://github.com/pytorch/vision/) that matches the PyTorch installation. | |
| Install them together at [pytorch.org](https://pytorch.org) to make sure of this. Note, please check | |
| PyTorch version matches that is required by Detectron2. | |
| - Detectron2: follow [Detectron2 installation instructions](https://detectron2.readthedocs.io/tutorials/install.html). | |
| - `pip install -e .` | |
| ### CUDA kernel for MSDeformAttn | |
| After preparing the required environment, run the following command to compile CUDA kernel for MSDeformAttn: | |
| `CUDA_HOME` must be defined and points to the directory of the installed CUDA toolkit. | |
| ```bash | |
| cd psalm/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops | |
| sh make.sh | |
| ``` | |
| ### Example conda environment setup | |
| ```bash | |
| conda create --name psalm python=3.10 -y | |
| conda activate psalm | |
| conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia | |
| # under your working directory | |
| git clone git@github.com:facebookresearch/detectron2.git | |
| cd detectron2 | |
| pip install . | |
| pip install git+https://github.com/cocodataset/panopticapi.git | |
| pip install git+https://github.com/mcordts/cityscapesScripts.git | |
| cd .. | |
| git clone https://github.com/zamling/PSALM.git | |
| cd PSALM | |
| pip install -e . | |
| pip install opencv-python addict | |
| cd psalm/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops | |
| sh make.sh | |