torch.mtia.set_device#
- torch.mtia.set_device(device)[source]#
Set the current device.
- Parameters
device (torch.device or int) – selected device. This function is a no-op if this argument is negative.
-
LearnGet Started
Run PyTorch locally or get started quickly with one of the supported cloud platforms
TutorialsWhats new in PyTorch tutorials
Learn the BasicsFamiliarize yourself with PyTorch concepts and modules
PyTorch RecipesBite-size, ready-to-deploy PyTorch code examples
Intro to PyTorch - YouTube SeriesMaster PyTorch basics with our engaging YouTube tutorial series
-
EcosystemTools
Learn about the tools and frameworks in the PyTorch Ecosystem
CommunityJoin the PyTorch developer community to contribute, learn, and get your questions answered
ForumsA place to discuss PyTorch code, issues, install, research
Developer ResourcesFind resources and get questions answered
Contributor Awards - 2024Award winners announced at this year's PyTorch Conference
-
Blogs & NewsPyTorch Blog
Catch up on the latest technical news and happenings
Community BlogStories from the PyTorch ecosystem
VideosLearn about the latest PyTorch tutorials, new, and more
Community StoriesLearn how our community solves real, everyday machine learning problems with PyTorch
EventsFind events, webinars, and podcasts
NewsletterStay up-to-date with the latest updates