Install

conda create -n RM python=3.10

conda activate RM

pip install torch torchvision

Run

python main