Update torch_verify.py

torch_verify.py

@@ -7,7 +7,7 @@ from PIL import Image
 # 加载预训练的ResNet50模型，并指定不使用GPU
 model = models.resnet50(pretrained=True)
 model.eval()  # 将模型设置为评估模式
-device = torch.device("cpu")  # 指定设备为CPU
+device = torch.device("cuda:0")  # 指定设备为CPU
 model.to(device)  # 将模型转移到CPU上
 
 # 图片预处理
@@ -45,4 +45,4 @@ with open(synset_words_path, 'r') as f:
 # 每行前去除空格和制表符，并根据索引获取类别名称
 class_names = [line.strip() for line in synset_words]
 predicted_class_name = class_names[predicted_class_idx+1]  # 注意索引可能从1开始，所以+1
-print(f"Predicted class name: {predicted_class_name}")  # 彭布罗克，彭布罗克威尔士柯基犬
\ No newline at end of file
+print(f"Predicted class name: {predicted_class_name}")  # 彭布罗克，彭布罗克威尔士柯基犬