在 python.py 文件中安装模块

Question

我正在按照 this 教程在 AWS sagemaker 上部署自定义 pytorch 模型。就我而言，我几乎没有安装某些模块的依赖项。

我的 inference.py 脚本中需要 pycocotools。我可以使用此 bash 命令轻松地将 pycocotool 安装在单独的笔记本中，

%%bash

pip -g install pycocotools

但是当我为部署创建我的端点时，我得到了一个错误，即 pycocotools 未定义。我的 inference.py 脚本中需要 pycocotools。我如何将其安装在 .py 文件中

Answer 1

在 inference.py 的开头添加这些行：

from subprocess import check_call, run, CalledProcessError
import sys
import os

# Since it is likely that you're going to run inference.py multiple times, this avoids reinstalling the same package:
if not os.environ.get("INSTALL_SUCCESS"):
    
    try:
        check_call(
        [ sys.executable, "pip", "install", "pycocotools",]
        )
    except CalledProcessError:
        run(
        ["pip", "install", "pycocotools",]
        )
    os.environ["INSTALL_SUCCESS"] = "True"

在 python.py 文件中安装模块

Installing modules inside python .py file

python

amazon-web-services

amazon-sagemaker