diff --git a/container.def b/container.def index cfdcf43ec5f1b98ec8e269fac7d9a0e1128291ba..9b1270dd0555441ef2252095ebd38162a8309778 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 0000000000000000000000000000000000000000..5610c40512071926769a90d452b2ddebd3f12aba --- /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