Skip to content

For Windows users #15

Closed
Closed
@jtydhr88

Description

@jtydhr88

I got it working on my Windows.
It might require Visual Studio Build Tools. However, I am not sure because my local already installed previously. If it needs, you can find from Visual Studio Build Tools.

Env

  1. WIndows 10
  2. Conda - Python 3.11
  3. 3090
  4. Cuda 12.1

Setup Steps

  1. conda create -n unique3d-py311 python=3.11
  2. conda activate unique3d-py311
  3. pip install torch torchvision torchaudio xformers --index-url https://download.pytorch.org/whl/cu121
  4. (Download the trition whl from https://huggingface.co/madbuda/triton-windows-builds, because I use python 311, so download whl build for py311 and install manually) pip install triton-2.1.0-cp311-cp311-win_amd64.whl
  5. pip install Ninja
  6. (Due the latest diffusers is not compatible to the code of repo, we MUST install this for diffusers manually) pip install diffusers==0.27.2
  7. pip install grpcio werkzeug tensorboard-data-server
  8. remove torch>=2.0.1, diffusers>=0.26.3, xformers, onnxruntime_gpu from requirement.txt
    image
  9. pip install -r requirements.txt
  10. pip uninstall onnxruntime
  11. pip uninstall onnxruntime-gpu
  12. (We MUST uninstall onnxruntime and onnxruntime-gpu first, then install this given version manually) pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

Runtime

  1. you need to create the output folder tmp\gradio under the running driver by yourself, for me, it is F:\tmp\gradio

  2. You will see the error of onnxruntime
    image
    but it would not effect the generation of the mesh (maybe?)

(Update) In order to fix the TensorRT error, you need to download TensorRT bundle for windows from https://github.com/NVIDIA/TensorRT, and configure TensorRT-10.0.1.6\lib in your PATH env variable. meanwhile also need to configure CUDA and cuDNN, please refer to github page for more details.

Quick test
image
image

Good luck~

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions