How can I install PyTorch framework in my environment?

Habana has dedicated PyTorch docker images for AmazonLinux2, Ubuntu18.04 and Ubuntu20.04. The docker images from the Habana Vault should be used to start a PyTorch environment on Habana.