如何验证 sagemaker 中是否安装了所有依赖项？

Question

我正在创建 sagemaker 端点并从 s3 存储桶加载预训练模型。模型 -> model.tar.gz 文件具有此处记录的目录结构，https://sagemaker.readthedocs.io/en/stable/frameworks/pytorch/using_pytorch.html#model-directory-structure

model.tar.gz/
|- model.pth
|- code/
  |- inference.py
  |- requirements.txt  # only for versions 1.3.1 and higher

我在 requirements.txt 中添加了一些依赖项，有没有办法验证所有依赖项都已正确安装？

Answer 1

无法访问或通过 SSH 进入您部署的运行机器。因此，一种方法是在“inference.py”内的 model_fn 中断言依赖项的版本，如下所示。

如果您的 requirements.txt 看起来像这样：

numpy==1.20.3
pandas==1.3.4

获取版本并在 `model_fn 中声明它们，如下所示：

import os

### your other code ###

def model_fn(model_dir):
    # assuming you have numpy and pandas
    assert os.popen("python3 -m pip freeze | grep -E 'numpy|pandas'").read() == 'numpy==1.20.3\npandas==1.3.4\n'
    ### your other code ###
    return xxxx

### your other code ###

如何验证 sagemaker 中是否安装了所有依赖项？

How to verify all the dependencies are installed in sagemaker?

serverless

amazon-sagemaker