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