Course contentsShow
Machine Learning and Deep Learning
Lesson 811 of 3,53819. PyTorch FundamentalsPro lesson

Moving Models Between Devices: CPU and GPU

Use .to(device) and .cuda() to move entire models between CPU and GPU, including all parameters and buffers.

This lesson is for subscribers

You've completed the free preview. Subscribe to unlock every lesson in every course.