fix: don't reinstall PyTorch (already installed)
This commit is contained in:
1 parent
3b771c0db5
commit
79aa1e4061
1 file changed
+1
-4
@@ -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