From 23dd617ac9fcd939fa67cd8f12687920a8fb7720 Mon Sep 17 00:00:00 2001
From: Luke Smith <smithl@tacc.utexas.edu>
Date: Thu, 20 Feb 2025 14:37:52 -0600
Subject: [PATCH] Re-building Docker container due to bug in xformers

---
 Dockerfile        |   2 +-
 TRELLIS/.DS_Store | Bin 6148 -> 6148 bytes
 TRELLIS/setup.sh  |   6 +++---
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 933e798..85caff7 100644
--- a/Dockerfile
+++ b/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
diff --git a/TRELLIS/.DS_Store b/TRELLIS/.DS_Store
index 2a76bc80fe026302d21e78c96db7cf373b889ff6..ecca33936507dc14f25a4a93faaabaa2043172f7 100644
GIT binary patch
delta 46
zcmZoMXffC@pN(<n<OOVn+>+JR7RE+83dTkTlV#au7`rF$XOU;@nH<Hwb~78tKYjpP
CWe$n}

delta 47
zcmZoMXffC@pN(<X<OOVnJW|!wrY43u3Wg>orjupZWf*%V?`M%`?42CRzIHP^$6tN`
DXc!K|

diff --git a/TRELLIS/setup.sh b/TRELLIS/setup.sh
index 33fe7d8..9a66ce7 100755
--- a/TRELLIS/setup.sh
+++ b/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