Hi all, Has anyone had any success running

pytorch 1.13.0

with GPU on Metaflow? I tried looking through this thread (https://github.com/Netflix/metaflow/issues/250), but didn't seem to work for me. The batch job is using

p2.xlarge