Its done.Just install according to the official website.

  
      
      pytorch.org
  
  
    

PyTorch 55

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.