There is a pinned topic for your situation:
In this post, there are installation steps and Docker images about PyTorch.
Best wishes