You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

无法在无GPU环境加载darts v0.22.0模型的求助

Darts v0.22.0模型GPU转CPU部署报错问题

近几个月我一直在使用darts v0.22.0,此前在GPU机器上训练模型并通过model.save()保存。现在尝试部署到仅CPU的机器上,使用以下代码将模型加载到CPU:

model = torch.load(home + "model.txt.ckpt", map_location=torch.device('cpu'))

模型加载成功,但调用model.historical_forecasts()时出现如下报错,看起来模型仍在寻找GPU:

未找到支持的GPU后端!Traceback (most recent call last):
File "/code/central/temp/onykz6rv/968/model/darts_class.py", line 171, in run_testing
preds = model.historical_forecasts(series=lst_test[t], past_covariates=lst_test_cov[t], retrain=False, forecast_horizon=horizon)
File "/root/miniconda3/lib/python3.9/site-packages/darts/utils/utils.py", line 172, in sanitized_method
return method_to_sanitize(self, *only_args.values(), **only_kwargs)
File "/root/miniconda3/lib/python3.9/site-packages/darts/models/forecasting/forecasting_model.py", line 500, in historical_forecasts
forecast = self._predict_wrapper(
File "/root/miniconda3/lib/python3.9/site-packages/darts/models/forecasting/forecasting_model.py", line 1228, in _predict_wrapper
return self.predict(
File "/root/miniconda3/lib/python3.9/site-packages/darts/utils/torch.py", line 112, in decorator
return decorated(self, *args, **kwargs)
File "/root/miniconda3/lib/python3.9/site-packages/darts/models/forecasting/torch_forecasting_model.py", line 1051, in predict
predictions = self.predict_from_dataset(
File "/root/miniconda3/lib/python3.9/site-packages/darts/utils/torch.py", line 112, in decorator
return decorated(self, *args, **kwargs)
File "/root/miniconda3/lib/python3.9/site-packages/darts/models/forecasting/torch_forecasting_model.py", line 1178, in predict_from_dataset
self._setup_trainer(trainer=trainer, verbose=verbose, epochs=self.n_epochs)
File "/root/miniconda3/lib/python3.9/site-packages/darts/models/forecasting/torch_forecasting_model.py", line 457, in _setup_trainer
self._init_trainer(trainer_params=trainer_params, max_epochs=epochs)
File "/root/miniconda3/lib/python3.9/site-packages/darts/models/forecasting/torch_forecasting_model.py", line 471, in _init_trainer
return pl.Trainer(**trainer_params_copy)
File "/root/miniconda3/lib/python3.9/site-packages/pytorch_lightning/utilities/argparse.py", line 340, in insert_env_defaults
return fn(self, **kwargs)
File "/root/miniconda3/lib/python3.9/site-packages/pytorch_lightning/trainer/trainer.py", line 414, in init
self._accelerator_connector = AcceleratorConnector(
File "/root/miniconda3/lib/python3.9/site-packages/pytorch_lightning/trainer/connectors/accelerator_connector.py", line 206, in init
self._accelerator_flag = self._choose_gpu_accelerator_backend()
File "/root/miniconda3/lib/python3.9/site-packages/pytorch_lightning/trainer/connectors/accelerator_connector.py", line 512, in _choose_gpu_accelerator_backend
raise MisconfigurationException("No supported gpu backend found!")
lightning_lite.utilities.exceptions.MisconfigurationException: No supported gpu backend found!

我尝试使用darts v0.23.1加载模型,但遇到了darts encoders包变更相关的错误。若无法解决该问题,我的大量工作将丢失,恳请各位提供帮助。
谢谢

内容的提问来源于stack exchange,提问作者Adeel Hashmi

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.29 06:30:34