pytorch 创建在gpu的张量

在PyTorch中，为了在GPU上创建张量，你需要先确认设备是否可用并且已经启用GPU。你可以使用`torch.cuda.is_available()`函数检查GPU是否存在，然后通过`torch.device('cuda')`来指定GPU作为计算设备。创建在GPU的张量的基本步骤如下：

1. 导入所需的库：

import torch

2. 检查并选择GPU：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

3. 使用`to(device)`将Python数据类型转换为GPU张量：

x = torch.tensor([1, 2, 3], dtype=torch.float).to(device)

在这个例子中，`x`就是一个存储在GPU上的浮点数张量。如果你想直接创建一个已经在GPU上的张量，可以使用`torch.randn()`或`torch.rand()`等生成随机数的方法，并加上`.to(device)`。