配置环境-insightface-torch

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

2.安装pytorch:

我的cuda 是 11.3

配置环境-insightface-torch_第1张图片

然后进入 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-torch_第2张图片

配置环境-insightface-torch_第3张图片

安装 insightface

pip install insightface

安装onnxruntime-gpu

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

你可能感兴趣的:(人脸识别,计算机视觉,python)