SageMaker Neo PyTorch 1.0.0

【问题标题】：SageMaker Neo PyTorch 1.0.0SageMaker Neo PyTorch 1.0.0
【发布时间】：2023-03-23 23:20:02
【问题描述】：

我已将 SageMaker pytorch_36 内核中的 torch 版本更新为 torch 版本 1.0.0。然后我尝试运行示例笔记本pytorch_torchvision_neo.ipynb，同时将framework_version 更改为1.0.0。 Neo 编译然后失败。

知道为什么它不适用于 1.0.0 吗？控制台错误消息实际上告诉我要确保我使用的是 1.0.0，但示例笔记本似乎只适用于 0.4.0。

【问题讨论】：

标签： python pytorch amazon-sagemaker

【解决方案1】：

Sagemaker 笔记本预装了 pytorch-1.1.0。但是模型编译服务需要pytorch-0.4.0或pytorch-1.0.1保存的模型

问题的解决方案：

# 1. do not install `pytorch-cpu` and `torchvision-cpu`.

# 2. Downgrade pytorch version to 1.0.1
!conda install -y pytorch=1.0.1 -c pytorch 

# 3. import pytorch and check that version is 1.0.1 (but not 1.1.0)
import torch 
torch.__version__

继续运行笔记本步骤：https://github.com/awslabs/amazon-sagemaker-examples/blob/master/sagemaker_neo_compilation_jobs/pytorch_torchvision/pytorch_torchvision_neo.ipynb

【讨论】：