CivArchive
    ← All articles
    Published July 18, 2024by BGCreator

    Simple method to use CUDA 12.1 and DWPose/Onnxruntime-GPU

    1K views5 reactions2 comments on CivitAI5 collected
    onnxruntimedwposecomfyuigeneration guide

    For those wishing to use DWPOSE or Onnxruntime with cuda 12.1 follow the following steps:

    • UNINSTALL OLD VERSION OF ONNXRUNTIME-GPU: pip uninstall onnxruntime-gpu

    1. pip install torch==2.3.0+cu121 torchvision==0.18.0+cu121 torchaudio==2.3.0+cu121 --extra-index-url https://download.pytorch.org/whl/cu121

    2. pip install onnx onnxruntime

    3. pip install onnxruntime-gpu==1.17.1 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

    For portable version:

    (thanks to Rvage for reporting)

    That's all!