Skip to content
Snippets Groups Projects
Commit df938970 authored by Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)'s avatar Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)
Browse files

imageio version check

parent c5645d94
No related branches found
No related tags found
No related merge requests found
Pipeline #102132 failed
...@@ -2,6 +2,9 @@ ...@@ -2,6 +2,9 @@
BootStrap: docker BootStrap: docker
From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
%files
env.yaml /workspace/env.yaml
%post %post
# Install special software or additions # Install special software or additions
apt-get update -y # Good practice, update the package database. apt-get update -y # Good practice, update the package database.
...@@ -11,6 +14,7 @@ From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 ...@@ -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. apt-get autoremove -y && apt-get clean && rm -rf /var/lib/apt/lists/* # keep the Docker image as small as possible.
# Install Miniconda # Install Miniconda
cd /workspace
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
bash miniconda.sh -b -p /opt/conda bash miniconda.sh -b -p /opt/conda
rm miniconda.sh rm miniconda.sh
...@@ -21,15 +25,9 @@ From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 ...@@ -21,15 +25,9 @@ From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
export TCNN_CUDA_ARCHITECTURES="75;80;86" export TCNN_CUDA_ARCHITECTURES="75;80;86"
. /opt/conda/etc/profile.d/conda.sh # Non - interactive sessions we need this . /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 env create -f env.yaml
conda clean -afy conda clean -afy
cd ..
rm -rf gala
%environment %environment
# Set some environment vars # Set some environment vars
export PATH=/opt/conda/bin:$PATH export PATH=/opt/conda/bin:$PATH
......
env.yaml 0 → 100644
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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment