Apple unified memory limits with PyTorch

[sanescope]

Posting this in case it saves someone else a little time.

If you were starting from scratch today for this use case, what would you choose first and what would you ignore?

I'm especially interested in mistakes worth avoiding.

[turboyard94]

I'd start with the simpler stack.

[dockridge]

Ran a quick check on my box. The issue went away once I changed one setting and restarted the stack cleanly.