| # Install CUDA 11.8 builds of PyTorch, TorchVision, and TorchAudio | |
| 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 all other dependencies | |
| pip install --no-cache-dir -r requirements.txt | |