I am not sure about it. But for me the 11.1 works.
check here
  
      
      pytorch.org
  
  
    

PyTorch 3

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