Error: The operator 'aten::grid_sampler_3d' is not currently implemented for the MPS device.

[deanocean]

I just got the following error while running. I am using Macbook Pro

「The operator 'aten::grid_sampler_3d' is not currently implemented for the MPS device. If you want this op to be considered for addition please comment on pytorch/pytorch#141287 and mention use-case, that resulted in missing op as well as commit hash b86416178253fcc5551617f17403a352d91e9655. As a temporary fix, you can set the environment variable PYTORCH_ENABLE_MPS_FALLBACK=1 to use the CPU as a fallback for this op. WARNING: this will be slower than running natively on MPS.」