diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 0a19537..35a4be8 100755
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -30,8 +30,8 @@ python3 -m venv venv
 source venv/bin/activate
 pip install --upgrade pip
 
-echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..."
-pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
+echo "[2/4] Install PyTorch with CUDA 13.0 (RTX 5090 compatible)..."
+pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130
 
 echo "[3/4] Install training dependencies..."
 pip install transformers datasets trl peft accelerate bitsandbytes deepspeed