Re-building Docker container due to bug in xformers

Dockerfile

@@ -10,7 +10,7 @@ RUN chmod ugo+wrx /code/TRELLIS
 
 ENV PATH="/code:$PATH"
 ENV PATH="/code/TRELLIS:$PATH"
-ENV CUDA_HOME=/usr/local/cuda-12
+ENV CUDA_HOME=/usr/local/cuda-12.2
 ENV LIBRARY_PATH="${CUDA_HOME}/lib64"
 ENV C_INCLUDE_PATH="${CUDA_HOME}/include"
 ENV LD_LIBRARY_PATH=/usr/local/cuda-12/lib64:$LD_LIBRARY_PATH

TRELLIS/setup.sh

@@ -64,7 +64,7 @@ if [ "$HELP" = true ] ; then
 fi
 
 if [ "$NEW_ENV" = true ] ; then
-    pip3 install torch==2.4.0 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu121
+    pip3 install torch==2.4.0 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu118
 fi
 
 # Get system information
@@ -107,7 +107,7 @@ fi
 
 
 
-pip3 install xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu121
+pip3 install xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu118
 # if [ "$XFORMERS" = true ] ; then
 #     # install xformers
 #     if [ "$PLATFORM" = "cuda" ] ; then
@@ -278,7 +278,7 @@ if [ "$VOX2SEQ" = true ] ; then
 fi
 
 
-pip3 install spconv-cu120
+pip3 install spconv-cu118
 
 # if [ "$SPCONV" = true ] ; then
 #     # install spconv