您现在的位置是：首页 >技术教程 >facenet-pytorch相关的版本对齐网站首页技术教程

CPU版本

打开命令行或AnacondaPrompt（没本质区别），输入：

pip install numpy==1.24.3
pip install tensorflow==2.15.0       # 用于活体检测等可选功能
pip install facenet-pytorch==0.3.0
pip install torch==2.0.1
pip install torchvision==0.15.2
pip install opencv-python==4.7.0.72

GPU版本

# 如果你用 CUDA 11.8
 

pip install numpy==1.24.3
pip install tensorflow==2.15.0
pip install facenet-pytorch==0.3.0
pip install torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
pip install torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
pip install opencv-python==4.7.0.72

# 或使用CUDA 11.7

# 核心库版本配置（GPU版）
pip install numpy==1.23.5             # 必须使用1.23.x系列保证CUDA兼容
pip install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
pip install torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
pip install torchaudio==0.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117
pip install facenet-pytorch==0.3.0    # 需要与PyTorch 1.13+兼容
pip install tensorflow==2.10.1        # 最后一个原生支持CUDA 11.7的TF版本

# 辅助库版本
pip install opencv-python==4.6.0.66   # 兼容CUDA内存管理
pip install scikit-learn==1.1.3       # 特征比对相关
pip install pillow==9.4.0             # 图像处理
pip install tqdm==4.64.1              # 进度条显示