Create setup.sh

setup.sh

@@ -0,0 +1,9 @@
+#!/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