Update setup_blender.sh

setup_blender.sh

@@ -4,7 +4,7 @@ set -e
 # --- Configuration ---
 BLENDER_VERSION="4.2.0"
 BLENDER_MAJOR_MINOR="4.2"
-BLENDER_PYTHON_VERSION="python3.11"
+BLENDER_PYTHON_VERSION="python3.11" # Should match the -dev package (e.g., python3.11-dev)
 BLENDER_TARBALL="blender-${BLENDER_VERSION}-linux-x64.tar.xz"
 BLENDER_URL="https://download.blender.org/release/Blender${BLENDER_MAJOR_MINOR}/blender-${BLENDER_VERSION}-linux-x64.tar.xz"
 INSTALL_DIR="/opt/blender-${BLENDER_VERSION}-linux-x64"
@@ -27,9 +27,27 @@ SPCONV_PACKAGE="spconv-${TARGET_CUDA_VERSION_SHORT}"
 # --- Set Environment Variables for Build ---
 export CUDA_HOME=${CUDA_HOME:-/usr/local/cuda}
 export PATH="${CUDA_HOME}/bin:${PATH}"
-export MAX_JOBS=4
+export MAX_JOBS=4 # For compilation processes like flash-attn
+# ** FIX: Explicitly add Python 3.11 include path for C/C++ header files **
+# This assumes python3.11-dev is installed via packages.txt, providing headers here.
+PYTHON_INCLUDE_DIR="/usr/include/python${BLENDER_PYTHON_VERSION#python}" # Extracts "3.11" from "python3.11"
+export CPATH="${PYTHON_INCLUDE_DIR}:${CPATH}"
+export C_INCLUDE_PATH="${PYTHON_INCLUDE_DIR}:${C_INCLUDE_PATH}"
+export CPLUS_INCLUDE_PATH="${PYTHON_INCLUDE_DIR}:${CPLUS_INCLUDE_PATH}"
+
 echo "Using CUDA_HOME=${CUDA_HOME}"
 echo "Updated PATH: ${PATH}"
+echo "CPATH set to: ${CPATH}"
+echo "C_INCLUDE_PATH set to: ${C_INCLUDE_PATH}"
+echo "CPLUS_INCLUDE_PATH set to: ${CPLUS_INCLUDE_PATH}"
+echo "Expected Python include directory: ${PYTHON_INCLUDE_DIR}"
+# Check if the Python.h file actually exists where we expect it
+if [ -f "${PYTHON_INCLUDE_DIR}/Python.h" ]; then
+    echo "Found Python.h at ${PYTHON_INCLUDE_DIR}/Python.h"
+else
+    echo "WARNING: Python.h NOT FOUND at ${PYTHON_INCLUDE_DIR}/Python.h. Ensure python${BLENDER_PYTHON_VERSION#python}-dev is in packages.txt and installed."
+fi
+
 
 # --- Download and Extract Blender ---
 echo "Downloading Blender ${BLENDER_VERSION}..."
@@ -74,12 +92,14 @@ echo "1. Installing PyTorch ${TORCH_VERSION} and Torchvision ${TORCHVISION_VERSI
     --index-url ${TORCH_INDEX_URL}
 
 echo "2. Installing dependencies from ${UNIRIG_REQS_FILE_IN_SPACE} (includes bpy, flash_attn)..."
+# flash-attn compilation might also benefit from CPATH
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir -r "${UNIRIG_REQS_FILE_IN_SPACE}"
 
 echo "3. Installing ${SPCONV_PACKAGE}..."
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir ${SPCONV_PACKAGE}
 
 echo "4. Installing torch-scatter and torch-cluster..."
+# The CPATH, C_INCLUDE_PATH, CPLUS_INCLUDE_PATH set above should help this step
 "${BLENDER_PY_EXEC}" -m pip install --no-cache-dir \
     torch-scatter \
     torch-cluster \