需要在基于python pytorch的代码中将GPU选项更改为CPU

import torch

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

model = YourModel().to(device)

input_data = input_data.to(device)

model.eval()

with torch.no_grad():
    output = model(input_data)

output = output.cpu()

import torch

# 检查系统是否支持GPU，并设置设备
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

# 加载模型并将其移动到所选设备上
model = YourModel().to(device)

# 将输入数据移动到所选设备上
input_data = input_data.to(device)

# 将模型设置为评估模式
model.eval()

# 禁用梯度计算以加快推理速度
with torch.no_grad():
    output = model(input_data)

# 将输出数据移动回CPU
output = output.cpu()