在pytorch中,读取GPU上张量的数值 (数据从GPU到CPU) 的几种常用方法

1、.cpu() 方法:
        使用 .cpu() 方法可以将张量从 GPU 移动到 CPU。这是一种简便的方法,常用于在进行 CPU 上的操作之前将数据从 GPU 取回

import torch

# 在 GPU 上创建一个张量
gpu_tensor = torch.tensor([1, 2, 3], device='cuda')

# 将 GPU 上的张量移动到 CPU
cpu_tensor = gpu_tensor.cpu()


# 打印输出
print("GPU Tensor:", gpu_tensor)
print("CPU Tensor:", cpu_tensor)
GPU Tensor: tensor([1,2,3],device='cuda:')
CPU Array: tensor([1,2,3])

2、.to('cpu') 方法:
        使用 .to('cpu') 方法也可以将张量移动到 CPU。这是一个通用的设备转移方法,可以指定目标设备和其他参数。

import torch

# 在 GPU 上创建一个张量
gpu_tensor = torch.tensor([1, 2, 3], device='cuda')

# 将 GPU 上的张量移动到 CPU
cpu_tensor = gpu_tensor.to('cpu')

# 打印输出
print("GPU Tensor:", gpu_tensor)
print("CPU Tensor:", cpu_tensor)
GPU Tensor: tensor([1,2,3],device='cuda:')
CPU Array: tensor([1,2,3])

3、.numpy() 方法:
        使用 .numpy() 方法将 GPU 上的张量转换为 NumPy 数组。这个方法实际上是先将张量移动到 CPU,然后转换为 NumPy 数组。

import torch

# 在 GPU 上创建一个张量
gpu_tensor = torch.tensor([1, 2, 3], device='cuda')

# 将 GPU 上的张量移动到 CPU,并转换为 NumPy 数组
cpu_array = gpu_tensor.cpu().numpy()

# 打印输出
print("GPU Tensor:", gpu_tensor)
print("CPU Array:", cpu_array)
GPU Tensor: tensor([1,2,3],device='cuda:')
CPU Array: array([1,2,3])

4、.tolist() 方法:
        使用 .tolist() 方法将张量转换为 Python 列表。

import torch

# 在 GPU 上创建一个张量
gpu_tensor = torch.tensor([1, 2, 3], device='cuda')

# 将张量转换为 Python 列表
python_list = gpu_tensor.tolist()

# 打印输出
print("GPU Tensor:",gpu_tensor)
print("\nPython_list:",python_list)
GPU Tensor: tensor([1,2,3],device='cuda:')
Python_list: [1,2,3]

5、.item() 方法:
        如果张量只包含一个元素,可以使用 .item() 方法直接获取该元素的 Python 数值。

import torch

# 在 GPU 上创建一个张量
gpu_tensor = torch.tensor(3, device='cuda')

# 获取张量的数值
value = gpu_tensor.item()

# 打印输出
print("GPU Tensor:", gpu_tensor)
print("Value:", value)
GPU Tensor: tensor(3,device='cuda:')
Value: 3

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2023-12-30 10:52:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2023-12-30 10:52:02       100 阅读
  3. 在Django里面运行非项目文件

    2023-12-30 10:52:02       82 阅读
  4. Python语言-面向对象

    2023-12-30 10:52:02       91 阅读

热门阅读

  1. docker -v 和docker --device 有什么区别

    2023-12-30 10:52:02       56 阅读
  2. 自动化测试:selenium总结

    2023-12-30 10:52:02       61 阅读
  3. 如何准备 Azure 信息保护云退出计划

    2023-12-30 10:52:02       55 阅读
  4. 【Linux】Ubuntu22.04版本下实现gcc版本的快速切换

    2023-12-30 10:52:02       66 阅读
  5. PyTorch中的 Dataset、DataLoader 和 enumerate()

    2023-12-30 10:52:02       56 阅读
  6. Linux添加一个指令代替指定指令

    2023-12-30 10:52:02       56 阅读
  7. 79. Word Search

    2023-12-30 10:52:02       66 阅读
  8. 蓝桥杯python比赛历届真题99道经典练习题 (8-12)

    2023-12-30 10:52:02       51 阅读