From 28d0a956fd64fb250577f43230179df481167e25 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Tue, 30 Jun 2026 16:58:11 -0400
Subject: [PATCH] fix: update PyTorch to cu124 for RTX 5090, fix total_memory
 typo

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

diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 10154db..26f20ff 100644
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -26,8 +26,8 @@ python3 -m venv venv
 source venv/bin/activate
 pip install --upgrade pip
 
-echo "[2/4] Install PyTorch with CUDA..."
-pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..."
+pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
 
 echo "[3/4] Install training dependencies..."
 pip install transformers datasets trl peft accelerate bitsandbytes
@@ -39,7 +39,8 @@ python3 -c "
 import torch
 if torch.cuda.is_available():
     print(f'  ✓ GPU: {torch.cuda.get_device_name(0)}')
-    print(f'  ✓ VRAM: {torch.cuda.get_device_properties(0).total_mem / 1e9:.1f} GB')
+    props = torch.cuda.get_device_properties(0)
+    print(f'  ✓ VRAM: {props.total_memory / 1e9:.1f} GB')
 else:
     print('  ✗ No GPU detected')
     print('  Please ensure CUDA drivers are installed')