File size: 306 Bytes
0521fcf |
1 2 3 4 5 6 7 8 9 10 |
#!/usr/bin/env bash
# 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
|