Hey @SamTse can you tell us what CUDA version you are on?
Also by any chance can you check that this is a problem with source install:

  
      
      GitHub
  
  
    

pytorch/pytorch 39

pytorch - Tensors and Dynamic neural networks in Python  with strong GPU acceleration