| echo "🔧 Starting custom setup..." | |
| # Install numpy first to avoid conflicts | |
| echo "📦 Installing NumPy..." | |
| pip install --no-cache-dir numpy==1.24.3 | |
| 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!" | |