diff --git a/src/model_trainer.py b/src/model_trainer.py
index 71385dd..12769ed 100644
--- a/src/model_trainer.py
+++ b/src/model_trainer.py
@@ -500,11 +500,15 @@ class ModelTrainer:
                     model.fit(X_train, y_train)
                     
                 elif model_type == 'pytorch':
-                    # 训练PyTorch模型
-                    train_dataset = EquipmentDataset(X_train, y_train)
-                    train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
-                    training_success = self.train_model(train_loader)
-                    if not training_success:
+                    # 训练PyTorch模型（如未安装则跳过）
+                    try:
+                        train_dataset = EquipmentDataset(X_train, y_train)
+                        train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
+                        training_success = self.train_model(train_loader)
+                        if not training_success:
+                            continue
+                    except ImportError as e:
+                        logger.warning(f"PyTorch not available, skipping: {e}")
                         continue
                 
                 # 评估模型性能