Go to pytorch website, and choose the version which satisfies your cuda version

  
      
      pytorch.org
  
  
    

PyTorch 140

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


  
  
    
    
  
  


cu100 = cuda 10.0
pip3 uninstall torch
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl