cloth-splatting.def

Bootstrap: docker
From: nvidia/cuda:12.0.0-devel-ubuntu22.04


%environment
# Environment variables for the NVIDIA container runtime
export NVIDIA_VISIBLE_DEVICES=all
export NVIDIA_DRIVER_CAPABILITIES=graphics,utility,compute

%post
# Update and install packages
apt-get update
DEBIAN_FRONTEND=noninteractive apt-get install -y \
    git python3-dev python3-pip \
    libglib2.0-0 \
    libglvnd0 \
    libgl1 \
    libglx0 \
    libegl1 \
    libxext6 \
    libx11-6 \
    cmake build-essential libgl1-mesa-dev freeglut3-dev libglfw3-dev libgles2-mesa-dev
rm -rf /var/lib/apt/lists/*

# Install PyTorch and other Python packages
pip install torch==2.2.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install torch_geometric
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cu121.html
pip install mmcv==1.6.0 matplotlib argparse lpips plyfile imageio-ffmpeg h5py imageio natsort numpy wandb open3d seaborn trimesh spatialmath-python pybind11 gym moviepy shapely



%runscript
# Default command to run
exec python3 "$@"