如何解决Kaggle环境中Facenet-pytorch的安装问题？

问题详情 #

在Kaggle环境运行代码时，导入facenet_pytorch的MTCNN模块报错，已尝试通过本地whl文件安装，但仍无法找到该模块。

使用的安装及导入代码 #

%%capture
!pip install /kaggle/input/facenet-pytorch-vggface2/facenet_pytorch-2.2.7-py3-none-any.whl

from facenet_pytorch import MTCNN
import cv2
from PIL import Image
import numpy as np
from matplotlib import pyplot as plt
from tqdm.notebook import tqdm

报错信息 #

ModuleNotFoundError                       Traceback (most recent call last)
/tmp/ipykernel_27/981336495.py in <module>
----> 1 from facenet_pytorch import MTCNN
      2 import cv2
      3 from PIL import Image
      4 import numpy as np
      5 from matplotlib import pyplot as plt

ModuleNotFoundError: No module named 'facenet_pytorch'

解决步骤 #

• 检查whl文件路径正确性
  先通过命令确认目标路径下的文件是否存在：

  !ls /kaggle/input/facenet-pytorch-vggface2/
  

  确保输出结果中包含facenet_pytorch-2.2.7-py3-none-any.whl，路径和文件名完全匹配。

• 取消%%capture查看安装日志
  移除%%capture后重新运行安装命令，查看是否有依赖缺失、权限报错等安装失败提示：

  !pip install /kaggle/input/facenet-pytorch-vggface2/facenet_pytorch-2.2.7-py3-none-any.whl
  

• 安装后重启内核
  Kaggle环境中安装新模块后，需重启内核才能让系统识别新安装的库。点击顶部菜单栏Kernel -> Restart Kernel，重启后再重新运行导入代码。

• 尝试从PyPI直接安装
  如果本地whl文件存在损坏或适配问题，可切换为官方源安装：

  !pip install facenet-pytorch
  

• 验证安装状态
  安装完成后，运行以下命令确认模块已成功安装：

  !pip show facenet-pytorch
  

  若输出包含模块版本、安装路径等信息，说明安装成功，再尝试导入MTCNN即可。

内容的提问来源于stack exchange，提问作者Faizkh