laoreja:




 smth:

import torch print(torch.version.cuda)


And I’m not using conda, just through pip install, in this way, is the CUDA also inbuilt?


Yes. There is a specific CPU version you can install though.
I think the only way to bring your own, you could compile and install it from source:

  
      
      pytorch.org
  
  
    

PyTorch 22

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