Uninstall current PyTorch, torchvision, and torchaudio

!pip uninstall torch torchvision torchaudio transformers -y

Install PyTorch 2.1 with CUDA 11.8 support

!pip install torch==2.7.1 torchvision==0.22.1 torchaudio==2.7.1 --index-url https://download.pytorch.org/whl/cu118

!pip install transformers==4.55.4