Running the demo notebook here as is with PyTorch v1.13.1 installed as specified in SageMaker references.
1 Like
pytorch_version='1.6',
This sample seems to assume torch 1.6. Either way, it’s a very old version…
It’s 2.5.1 now…
Even for stability, 2.2 is more common.
2.3 is a minefield.