From b760ba09e7f16ed0d5fc1abfd49f7af54d9d6b6b Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Tue, 30 Jun 2026 22:32:14 -0400
Subject: [PATCH] fix: use CUDA 13.0 nightly for RTX 5090 support

---
 train-on-this-server.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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