torch.cuda.is_available

Software / App

A PyTorch function to check if a CUDA-enabled GPU is available.

Mentioned in 1 video