torch.cuda — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.cuda¶ This package adds support for CUDA tensor types, that implement the same function as CPU tensors, but they utilize GPUs for computation. It is lazily initialized, so you can always import it, and use is_available() to determine if your system supports CUDA. CUDA semantics has more details about working with CUDA.