File size: 515 Bytes
0521fcf 55c5dda fc9eb64 55c5dda fc9eb64 0521fcf 55c5dda fc9eb64 0521fcf 55c5dda |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
#!/usr/bin/env bash
echo "🔧 Starting custom setup..."
# Install numpy first to avoid conflicts
pip install --no-cache-dir numpy==1.23.5
echo "📦 Installing CUDA-enabled PyTorch..."
# Install CUDA-enabled PyTorch
pip install --no-cache-dir \
torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 \
--index-url https://download.pytorch.org/whl/cu118
echo "📋 Installing remaining dependencies..."
# Install remaining dependencies
pip install --no-cache-dir -r requirements.txt
echo "✅ Setup complete!"
|