pytorch move tensor to cpu

Описание: Download this code from https://codegive.com
PyTorch is a popular deep learning framework that supports both CPU and GPU computations. In some cases, you may need to move your tensors between devices, such as moving a tensor from GPU to CPU. This tutorial will guide you through the process of moving a PyTorch tensor to the CPU using code examples.
Before we begin, make sure you have PyTorch installed. You can install it using:
Now, import PyTorch in your Python script or Jupyter Notebook:
Let's start by creating a PyTorch tensor on the GPU. This is typically done using the cuda() method:
In this example, we create a random tensor of shape (3, 3) on the GPU.
To move the tensor from the GPU to the CPU, you can use the to method with the target device set to cpu:
Now, the cpu_tensor contains the same data as gpu_tensor but is located in the CPU memory.
You can verify the device of a tensor using the device attribute. Let's check the devices of both tensors:
This will print the devices of both tensors, confirming that the cpu_tensor is now on the CPU.
Here's the complete code example:
This tutorial covers the basics of moving PyTorch tensors from GPU to CPU. Understanding tensor movement between devices is crucial for managing resources efficiently in deep learning applications.
ChatGPT