配置环境-insightface-torch

1. 创建环境：conda create -n insightface2 python=3.8

2.安装pytorch:

我的cuda 是 11.3

然后进入 pytorch  官网查找对应cuda 版本 pytorch 安装

建议使用 pip 

# CUDA 11.3
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

或者

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

安装 insightface

pip install insightface

安装onnxruntime-gpu

pip install onnxruntime-gpu -i https://mirror.baidu.com/pypi/simple