pytorch.org
  
  
    

PyTorch 202

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


  
  
    
    
  
  


Install via above link. It should work.