안녕하세요 오늘 포스트는 TRELLIS를 빌드하여 사용해보자 입니다.
우선 TRELLIS란 마이크로소프트에서 공개한 오픈소스 이미지 3D 변환 툴이고요
주어진 이미지나 텍스트에 대해서 3D 오브젝트를 생성해줍니다.
자 그럼 해당 툴을 빌드해봅시다.
cd <원하는 디렉터리>
git clone --recurse-submodules https://github.com/microsoft/TRELLIS.git
git submodule init
git submodule update --remote
git pull origin main
cd TRELLIS
conda create -n TRELLIS_BUILD -y python==3.10.16
conda env list
conda activate TRELLIS_BUILD
pip install torch==2.5.1 torchvision --index-url=https://download.pytorch.org/whl/cu124
pip install xformers==0.0.28.post3 --index-url=https://download.pytorch.org/whl/cu124
pip install pillow imageio imageio-ffmpeg tqdm easydict opencv-python-headless scipy ninja rembg onnxruntime trimesh xatlas pyvista pymeshfix igraph transformers
pip install git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
pip install https://github.com/bdashore3/flash-attention/releases/download/v2.7.1.post1/flash_attn-2.7.1.post1+cu124torch2.5.1cxx11abiFALSE-cp310-cp310-win_amd64.whl
3.11.3의 경우
pip install https://github.com/bdashore3/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu124torch2.5.1cxx11abiFALSE-cp311-cp311-win_amd64.whl
pip install kaolin -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124.html
git clone https://github.com/NVlabs/nvdiffrast.git ./tmp/extensions/nvdiffrast
pip install ./tmp/extensions/nvdiffrast
git clone --recurse-submodules https://github.com/JeffreyXiang/diffoctreerast.git ./tmp/extensions/diffoctreerast
pip install ./tmp/extensions/diffoctreerast
git clone https://github.com/autonomousvision/mip-splatting.git ./tmp/extensions/mip-splatting
pip install ./tmp/extensions/mip-splatting/submodules/diff-gaussian-rasterization/
xcopy .\extensions\vox2seq .\tmp\extensions\vox2seq /E /I
pip install ./tmp/extensions/vox2seq
pip3 install spconv-cu124
pip install gradio==4.44.1 gradio_litmodel3d==0.0.1
requirements.txt를 참고하세요(경로는 알맞게 바꿔서 쓰시면 됩니다 🙂
pip install -r requirements.txt
python==3.11.3 용 windows triton
pip install https://huggingface.co/madbuda/triton-windows-builds/resolve/main/triton-2.1.0-cp311-cp311-win_amd64.whl
해당 triton을 extensions에 다운 받습니다.
cd extensions
pip install ./triton-3.0.0-cp310-cp310-win_amd64.whl
해당 nvdiffrast도 똑같이 합니다.
pip install ./nvdiffrast-0.3.3-py3-none-any.whl
cd ..
pip install open3d==0.19.0
pip install plyfile==1.1.2
pip install git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
pip install numpy==1.26.4
pip install numba==0.61.2
pip install gradio==5.33.2
set ATTN_BACKEND=flash-attn
set SPCONV_ALGO=native
python ./app.py
