Shortcuts

Function at::cuda::is_available

Function Documentation

bool at::cuda::is_available()

CUDA is available if we compiled with CUDA, and there are one or more devices.

Returns true if at least one CUDA device is available.

If we compiled with CUDA but there is a driver problem, etc., this function will report CUDA is not available (rather than raise an error.)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources