---

venv/bin/pip uninstall -y torch torchvision torchaudio
venv/bin/pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu121