Update setup.sh

setup.sh

@@ -1,9 +1,9 @@
 #!/usr/bin/env bash
 
-# Install CUDA 11.8 builds of PyTorch, TorchVision, and TorchAudio
+# 1. 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 all other dependencies
+# 2. Install all other dependencies
 pip install --no-cache-dir -r requirements.txt