fix: don't reinstall PyTorch (already installed)
This commit is contained in:
@@ -10,10 +10,7 @@ python3 -m venv venv
|
||||
source venv/bin/activate
|
||||
pip install --upgrade pip
|
||||
|
||||
# Install dependencies
|
||||
echo "Installing PyTorch with CUDA 12.4..."
|
||||
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124
|
||||
|
||||
# Install training dependencies (PyTorch already installed)
|
||||
echo "Installing training dependencies..."
|
||||
pip install transformers datasets trl peft accelerate bitsandbytes deepspeed
|
||||
|
||||
|
||||
Reference in New Issue
Block a user