window10安装insightface、onnxruntime-gpu、视频换脸

从雍和宫走到电影学院 / 2024-04-23 / 原文

1. https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements

2.下载cuda_12.2.0_536.25_windows

3.下载cudnn cudnn-windows-x86_64-8.5.0.96_cuda11-archive

4.conda create -n onnx_test python=3.10

5.根据官方文档说明cuda12.x的版本安装onnx采用如下方式。

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

6.提示错误：Could not locate zlibwapi.dll. Please make sure it is in your library path!

https://www.winimage.com/zLibDll/ 下载

附：

Package                Version
---------------------- -----------
albumentations         1.4.4
annotated-types        0.6.0
certifi                2024.2.2
charset-normalizer     3.3.2
cmake                  3.29.2
colorama               0.4.6
coloredlogs            15.0.1
contourpy              1.2.1
cycler                 0.12.1
Cython                 3.0.10
easydict               1.13
flatbuffers            24.3.25
fonttools              4.51.0
humanfriendly          10.0
idna                   3.7
imageio                2.34.1
insightface            0.7.3
joblib                 1.4.0
kiwisolver             1.4.5
lazy_loader            0.4
matplotlib             3.8.4
mpmath                 1.3.0
networkx               3.3
numpy                  1.26.4
onnx                   1.16.0
onnxruntime-gpu        1.17.1
opencv-python          4.9.0.80
opencv-python-headless 4.9.0.80
packaging              24.0
pillow                 10.3.0
pip                    24.0
prettytable            3.10.0
protobuf               5.26.1
pydantic               2.7.0
pydantic_core          2.18.1
pyparsing              3.1.2
pyreadline3            3.4.1
python-dateutil        2.9.0.post0
PyYAML                 6.0.1
requests               2.31.0
scikit-image           0.23.2
scikit-learn           1.4.2
scipy                  1.13.0
setuptools             69.5.1
six                    1.16.0
sympy                  1.12
threadpoolctl          3.4.0
tifffile               2024.4.18
tqdm                   4.66.2
typing_extensions      4.11.0
urllib3                2.2.1
wcwidth                0.2.13
wheel                  0.43.0