yeah… You are right… Check this answer:

  
    
    
    Torch.cuda.is_available() returns False, nvidia-smi is working 
  
  
    Which CUDA version are you using? 
Based on the driver, you could only use CUDA<=9.2 as shown here. 
I would recommend to update the driver and the CUDA version (if you are using an older one).
  


You could install a previous version of PyTorch that supports CUDA-9.0…

  
      
      pytorch.org
  
  
    

PyTorch 45

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