How to install MPS backend on docker/podman container?

I’m trying to install pytorch with MPS backend - has anyone tried installing it on Docker?

Hey!

Did you follow the instructions in https://pytorch.org/get-started/locally/ to install inside your docker image?

Thanks @albanD How can I set up PyTorch in editable mode on my Macbookso that I can make code changes locally and have those changes immediately reflected when I use the library? I want to configure my environment for contributing to PyTorch. But online tutorials are very few including docs which are not very clear. I’m ideally looking at setting it up in a dev container. Example: GitHub - keras-team/keras-cv: Industry-strength Computer Vision workflows with Keras has a pretty neat dev container file making it very easy to setup.