Added Singularity recipe for StableViewSynthesis dependencies

1 year ago · bc8f0ebd72
parent 7179d026d5
commit bc8f0ebd72
1 changed files with 12 additions and 0 deletions
--- a/aicloud_slurm/stable_view_synthesis/StableViewSynthesis.def
+++ b/aicloud_slurm/stable_view_synthesis/StableViewSynthesis.def
@ -0,0 +1,12 @@
+Bootstrap: docker
+From: nvcr.io/nvidia/pytorch:21.10-py3
+
+%post
+    export DEBIAN_FRONTEND=noninteractive
+    apt-get update
+    apt-get install -y libeigen3-dev cmake
+    /opt/conda/bin/conda install scikit-image pillow pip
+    #/opt/conda/bin/conda install -c pyg pytorch-scatter pytorch-sparse pyg
+    /opt/conda/bin/pip install torch-scatter torch-sparse torch-geometric
+    /opt/conda/bin/conda install -c open3d-admin open3d
+    unset DEBIAN_FRONTEND