CoreML入门遇ModuleNotFoundError：coremltools模块缺失求解

问题详情 #

我是CoreML新手，在将文件转换为mlmodel时遇到问题。我的代码如下：

import coremltools

caffe_model = ('oxford102.caffemodel','deploy.prototxt')

labels = 'flower-labels.txt'

coreml_model = coremltools.converters.caffe.convert(
    caffe_model,
    class_label = labels,
    image_input_name = 'data'
)

coreml_model.save('FlowerClassifer.mlmodel')

执行python convert-script.py命令时，出现如下错误：

ModuleNotFoundError: No module named 'coremltools'

当前Python版本为3.9，使用Mac系统，网上显示coremltools要求Python2.7，已尝试各类网上解决方法仍未解决，请问该如何处理？

解决方法 #

• 更新版本认知
  旧版coremltools确实仅支持Python2.7，但从coremltools 4.0版本开始，已经兼容Python3.6及以上版本，完全支持你使用的Python3.9，你看到的是过时信息。

• 正确安装coremltools
  在Mac系统下，用对应Python3.9的pip命令安装：

  # 明确指定Python3.9的pip
  python3 -m pip install coremltools
  # 若系统pip已指向Python3.9，可直接用
  pip install coremltools
  

  注意不要使用Python2.7的pip执行安装，避免模块安装到错误环境。

• 验证环境与执行脚本
  安装完成后，在终端执行以下命令验证：

  python3 -c "import coremltools; print(coremltools.__version__)"
  

  若能正常输出版本号，说明安装成功。之后执行转换脚本时，需使用对应Python3.9的命令：

  python3 convert-script.py
  

  避免系统默认python命令指向Python2.7导致找不到模块。

• 虚拟环境隔离（推荐）
  为避免不同Python版本的环境冲突，可创建虚拟环境：

  # 创建Python3.9的虚拟环境
  python3 -m venv coreml-env
  # 激活虚拟环境
  source coreml-env/bin/activate
  # 安装coremltools
  pip install coremltools
  # 执行转换脚本
  python convert-script.py
  # 完成后退出虚拟环境
  deactivate
  

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