Skip to content

Fixes for setup.py and basic docker support #76

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
paroj wants to merge 3 commits into
base: main
Choose a base branch
from
+22 −3
Open

Fixes for setup.py and basic docker support #76

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
6b190bb
add missing dependencies for app.py
paroj Jan 5, 2026
4110348
fix build-isolation for flash-attn
paroj Jan 5, 2026
8cddb43
add Dockerfile
paroj Jan 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 19 additions & 0 deletions Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
# syntax=docker/dockerfile:1-labs
# docker buildx create --name gpubuilder --driver-opt "image=moby/buildkit:buildx-stable-1-gpu" --bootstrap
# docker buildx --builder gpubuilder build --allow device -t trellis --load .
FROM nvidia/cuda:12.9.1-cudnn-devel-ubuntu22.04
ARG DEBIAN_FRONTEND=noninteractive
RUN apt update && apt install -y g++ git cmake ninja-build \
python3 python-is-python3 python3-pip python3-venv sudo
RUN git clone -b main https://github.com/microsoft/TRELLIS.2.git --recursive
# create venv instead of conde (see --new-env in setup.sh)
RUN python3 -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"
RUN pip install -U wheel setuptools pip
RUN pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
WORKDIR /TRELLIS.2
ARG TORCH_CUDA_ARCH_LIST="8.0"
RUN --device=nvidia.com/gpu bash -c ". ./setup.sh --basic --flash-attn --nvdiffrast --nvdiffrec --cumesh --o-voxel --flexgemm"
EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"
CMD ["python", "app.py"]
6 changes: 3 additions & 3 deletions setup.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -77,16 +77,16 @@ if [ "$NEW_ENV" = true ] ; then
fi

if [ "$BASIC" = true ] ; then
pip install imageio imageio-ffmpeg tqdm easydict opencv-python-headless ninja trimesh transformers gradio==6.0.1 tensorboard pandas lpips zstandard
pip install imageio imageio-ffmpeg tqdm easydict opencv-python-headless ninja trimesh transformers gradio==6.0.1 tensorboard pandas lpips zstandard plyfile psutil
pip install git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
sudo apt install -y libjpeg-dev
sudo apt install -y libjpeg-dev libwebp-dev
pip install pillow-simd
pip install kornia timm
fi

if [ "$FLASHATTN" = true ] ; then
if [ "$PLATFORM" = "cuda" ] ; then
pip install flash-attn==2.7.3
pip install flash-attn==2.7.3 --no-build-isolation
elif [ "$PLATFORM" = "hip" ] ; then
echo "[FLASHATTN] Prebuilt binaries not found. Building from source..."
mkdir -p /tmp/extensions
Expand Down