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