diff --git a/runpod_setup.sh b/runpod_setup.sh
index 34b635d..1d6f301 100755
--- a/runpod_setup.sh
+++ b/runpod_setup.sh
@@ -83,6 +83,10 @@ else
     cd ..
 fi
 
+# Fix PyTorch version conflicts after SAM2 installation
+print_status "Fixing PyTorch version conflicts..."
+pip install torchaudio --upgrade --no-deps || print_error "Failed to upgrade torchaudio"
+
 # Download SAM2 checkpoints
 print_status "Downloading SAM2 checkpoints..."
 cd segment-anything-2/checkpoints