| # Install specific numpy version first to avoid conflicts | |
| pip install --no-cache-dir numpy==1.23.5 | |
| # 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 | |
| # Install remaining dependencies | |
| pip install --no-cache-dir -r requirements.txt | |