diff --git a/container.def b/container.def
index 9b1270dd0555441ef2252095ebd38162a8309778..ce1b0a763b6ca647ceb8ba34d78bd280a698c59e 100644
--- a/container.def
+++ b/container.def
@@ -1,38 +1,23 @@
 # Get the base image from nvidia/cuda Docker image.
 BootStrap: docker
-From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
-
-%files
-    env.yaml /workspace/env.yaml
+From: nvidia/cuda:12.8.1-cudnn-devel-ubuntu20.04
 
 %post
     # Install special software or additions
     apt-get update -y # Good practice, update the package database.
     apt-get install -y \
-        wget curl vim bzip2 git build-essential \
+        wget sshfs rsync curl vim bzip2 zip unzip git build-essential cmake \
         libopenblas-dev 
     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
-
-    # Code environment setup
-    export PATH=/opt/conda/bin:$PATH
-    export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6+PTX"
-    export TCNN_CUDA_ARCHITECTURES="75;80;86"
-    . /opt/conda/etc/profile.d/conda.sh # Non - interactive sessions we need this
-    
-    conda env create -f env.yaml
-    conda clean -afy
-    
 %environment
     # Set some environment vars
-    export PATH=/opt/conda/bin:$PATH
-    . /opt/conda/etc/profile.d/conda.sh
-    conda activate gala
+    export PATH=/mnt/fast/nobackup/users/ad02037/anaconda3/bin:$PATH
+    export CONDA_PREFIX="/mnt/fast/nobackup/users/ad02037/anaconda3/envs/gala"
+    export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6+PTX"
+    export TCNN_CUDA_ARCHITECTURES="75;80;86"
+    export CUDA_HOME=/usr/local/cuda
+    export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/lib64/stubs:$CONDA_PREFIX/lib:$LD_LIBRARY_PATH
 
 %labels
     # Add custom metadata to the container.
diff --git a/env.yaml b/env.yaml
deleted file mode 100644
index 5610c40512071926769a90d452b2ddebd3f12aba..0000000000000000000000000000000000000000
--- a/env.yaml
+++ /dev/null
@@ -1,29 +0,0 @@
-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