TRELLIS项目安装与配置指南

TRELLIS项目安装与配置指南

TRELLIS Official repo for paper "Structured 3D Latents for Scalable and Versatile 3D Generation". 项目地址: https://gitcode.com/gh_mirrors/trell/TRELLIS

1. 项目基础介绍

TRELLIS 是一个开源的3D资产生成模型，它能够接收文本或图像提示并生成高质量的三维资产，如辐射场、3D高斯分布和网格等。该项目使用的核心技术是统一的结构化潜在（SLAT）表示，这使得解码到不同的输出格式成为可能，同时配合优化的Rectified Flow Transformers作为强大的骨架网络。

主要编程语言：Python

2. 项目使用的关键技术和框架

• SLAT（Structured LATent）表示：允许模型将潜在空间中的数据解码成多种不同的3D表示。
• Rectified Flow Transformers：为SLAT优化的变换器网络，用于处理三维数据。
• CUDA Toolkit：用于GPU加速计算。
• Conda：用于管理项目环境和依赖。
• PyTorch：深度学习框架，用于模型的训练和推理。
• xformers, spconv, mipgaussian 等依赖库：用于支持特定操作和优化。

3. 项目安装和配置的准备工作

系统要求

• 操作系统：Linux（Windows用户可以参考相关文档，但未完全测试）
• 硬件：至少16GB内存的NVIDIA GPU（已验证NVIDIA A100和A6000 GPU）
• 软件：CUDA Toolkit（用于编译某些子模块），推荐使用Conda进行依赖管理，Python版本3.8或更高

安装步骤

1. 克隆项目仓库：

   git clone --recurse-submodules https://github.com/microsoft/TRELLIS.git
   cd TRELLIS
   

2. 创建新的Conda环境并安装依赖：

   根据你的CUDA版本和是否需要创建新环境，以下命令可能会有所不同。以下是一个基本示例：

   ./setup.sh --new-env --basic --xformers --flash-attn --diffoctreerast --spconv --mipgaussian --kaolin --nvdiffrast
   

   如果需要更详细的帮助信息，可以运行：

   ./setup.sh --help
   

   请注意以下几点：

   • --new-env 参数会创建一个新的Conda环境。
   • --basic 参数安装基本依赖。
   • 如果你的GPU不支持 flash-attn，可以去掉 --flash-attn 参数，并设置环境变量 ATTN_BACKEND=xformers。
   • 安装过程可能需要一些时间，请耐心等待。

3. 如果在安装过程中遇到问题，可以尝试逐个安装依赖项，或者打开一个 Issue 寻求帮助。

通过以上步骤，你应该能够成功安装 TRELLIS 项目，并进行后续的开发或研究工作。

TRELLIS Official repo for paper "Structured 3D Latents for Scalable and Versatile 3D Generation". 项目地址: https://gitcode.com/gh_mirrors/trell/TRELLIS