jetson nano上怎么让torch使用gpu而不使用cpu

可以通过以下代码将 PyTorch 切换到 GPU 模式：

import torch

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

# 将模型和数据移动到 GPU 上
model.to(device)
data = data.to(device)

# 在 GPU 上运行模型
output = model(data)

其中，`torch.cuda.is_available()` 用于检查是否有可用的 GPU，如果有，则使用 `cuda:` 作为设备名称，否则使用 `cpu`。然后，将模型和数据移动到 GPU 上，最后在 GPU 上运行模型即可。