Question

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

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

Answer 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

SageMaker Neo PyTorch 1.0.0

1 个答案: