pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'

# Manually install paddlepaddle-gpu by following this reference:
# https://www.paddlepaddle.org.cn/
python -m pip install paddlepaddle-gpu==2.6.1.post120 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html

# clone this project and run the following command
pip install -e .

python -m aim.scripts.prepare # Download weights for inference