pytorch

torch.cuda.is_available()