#!/bin/bash

# uninstall torch and torchaudio
uv pip uninstall torch torchaudio

# install torch and torchaudio with CUDA support
uv pip install torch~=2.7.1 torchaudio~=2.7.1 --index-url https://download.pytorch.org/whl/cu128