From df938970197aa07ced3786fd080e8d5e82128ab2 Mon Sep 17 00:00:00 2001
From: "Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)" <ayushi.dutta@surrey.ac.uk>
Date: Mon, 7 Apr 2025 18:37:23 +0000
Subject: [PATCH] imageio version check

---
 container.def | 12 +++++-------
 env.yaml      | 29 +++++++++++++++++++++++++++++
 2 files changed, 34 insertions(+), 7 deletions(-)
 create mode 100644 env.yaml

diff --git a/container.def b/container.def
index cfdcf43..9b1270d 100644
--- a/container.def
+++ b/container.def
@@ -2,6 +2,9 @@
 BootStrap: docker
 From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 
+%files
+    env.yaml /workspace/env.yaml
+
 %post
     # Install special software or additions
     apt-get update -y # Good practice, update the package database.
@@ -11,6 +14,7 @@ From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
     apt-get autoremove -y && apt-get clean && rm -rf /var/lib/apt/lists/* # keep the Docker image as small as possible.
     
     # Install Miniconda
+    cd /workspace
     wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
     bash miniconda.sh -b -p /opt/conda
     rm miniconda.sh
@@ -21,15 +25,9 @@ From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
     export TCNN_CUDA_ARCHITECTURES="75;80;86"
     . /opt/conda/etc/profile.d/conda.sh # Non - interactive sessions we need this
     
-    mkdir -p /workspace
-    cd /workspace
-    git clone https://github.com/snuvclab/gala.git
-    cd gala
     conda env create -f env.yaml
     conda clean -afy
-    cd ..
-    rm -rf gala
-
+    
 %environment
     # Set some environment vars
     export PATH=/opt/conda/bin:$PATH
diff --git a/env.yaml b/env.yaml
new file mode 100644
index 0000000..5610c40
--- /dev/null
+++ b/env.yaml
@@ -0,0 +1,29 @@
+name: gala
+channels:
+  - pytorch
+  - nvidia
+  - defaults
+dependencies:
+  - pytorch=1.13.0
+  - torchvision=0.14.0
+  - pytorch-cuda=11.7
+  - pip
+  - pip:
+    - tqdm
+    - ninja
+    - transformers
+    - accelerate
+    - diffusers==0.21.4
+    - open3d
+    - tetgen
+    - pymeshfix
+    - trimesh
+    - omegaconf
+    - triton
+    - mediapipe
+    - controlnet-aux==0.0.7
+    - imageio==2.19.3
+    - imageio-ffmpeg
+    - git+https://github.com/NVlabs/nvdiffrast.git
+    - git+https://github.com/facebookresearch/pytorch3d.git
+    - git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
\ No newline at end of file
-- 
GitLab