Sign In

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

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

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!

5

Comments