diff --git a/train-on-this-server.sh b/train-on-this-server.sh index 7f46d79..0e3d40f 100644 --- a/train-on-this-server.sh +++ b/train-on-this-server.sh @@ -27,7 +27,7 @@ source venv/bin/activate pip install --upgrade pip echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..." -pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 +pip install --force-reinstall 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