We used PyTorch 1.9.1 on Ubuntu 22.04 LTS with Anaconda Python 3.7.
-
[Optional but recommended] Create a new Conda environment.
conda create --name omni_seg python=3.7
And activate the environment.
conda activate omni_seg
-
Clone the Omni_Seg repo
-
Install the requirements
-
Install apex:
cd Omni_Seg/Omni_seg_pipeline_gpu/apex python3 setup.py install cd ..