PyTorch CUDA | Complete Guide on PyTorch CUDA
https://www.educba.com/pytorch-cudatorch.cuda.is_available() It is good to know about CUDA in the system, and the below commands help in the same. torch.cuda.current_device() torch.cuda.get_device_name(ID of the device) torch.cuda.memory_allocated(ID of the device) torch.cuda.memory_reserved(ID of the device) Cached memory can be released from CUDA using the following command.