Update setup.sh

setup.sh

@@ -1,5 +1,13 @@
 #!/usr/bin/env bash
 set -e
 
-# 2) Then install the rest
-pip install -r requirements.txt 
+# 1) Upgrade pip itself
+python3 -m pip install --upgrade pip
+
+# 2) Install PyTorch FIRST (CPU‐only here; switch to +cu118 for GPU)
+python3 -m pip install \
+    torch==2.0.1+cpu \
+    --extra-index-url https://download.pytorch.org/whl/cpu
+
+# 3) Now install everything else (this includes torchmcubes --no-binary)
+python3 -m pip install -r requirements.txt