Which virtual environment best for Hi3DGen project

Hi group,

I’m struggling with seting up environment for hi3dgen(GitHub - Stable-X/Stable3DGen: A Modular Framework for 3D Generation and Beyond [WIP]) project which generates 3d mesh from image.

The main requirements are torch,diffusers, gradio,accelerate,spconv,triton,transformers

What virtual env would you chose to achieve this? would it be python env, docker container or conda?

Thank you.

tj

Hi,
You can find some packages in jp6/cu126 index
such as torch, torchvision.

There may be no spconv-cu package and you may explore building it. Not sure whether it is supported on aarch64 architecture.