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

跨Mac与Linux加载PPO模型时预测方法卡顿问题求助

PPO模型跨平台加载后predict方法卡顿问题排查

在M2 Max芯片的Mac OS上使用stable-baselines3和OpenAI Gym训练PPO模型,为避免Mac过载,在搭载Intel Core I7的Zorin Linux机器上运行遗传算法。该模型在Mac上可正常使用,但在Linux机器加载后,调用predict方法时出现卡顿。

相关代码

Mac端模型保存代码

model.save(f"/Users/.../Desktop/Coding/Personal/Python/QuantTrading/Code/DRL/stupidcode/models/BTCBITOmark1")

Linux端模型加载代码

model = PPO.load(f'/home/michael/Desktop/Code/DRL/stupidcode/models/BTCBITOmark1.zip')

Linux端predict调用代码

self.model = model
observation = self.get_observation()
action = self.model.predict(observation, deterministic=True)[0]

环境信息

两台机器Python版本均为3.11.4,依赖版本基本一致。加载模型时添加print_system_info=True后输出的系统信息如下:

== CURRENT SYSTEM INFO ==
- OS: Linux-5.15.0-76-generic-x86_64-with-glibc2.31 # 83~20.04.1-Ubuntu SMP Wed Jun 21 20:23:31 UTC 2023
- Python: 3.11.4
- Stable-Baselines3: 2.0.0
- PyTorch: 2.0.1+cu117
- GPU Enabled: False
- Numpy: 1.25.1
- Cloudpickle: 2.2.1
- Gymnasium: 0.28.1
- OpenAI Gym: 0.26.2

== SAVED MODEL SYSTEM INFO ==
- OS: macOS-12.5.1-arm64-arm-64bit Darwin Kernel Version 21.6.0: Wed Aug 10 14:28:23 PDT 2022; root:xnu-8020.141.5~2/RELEASE_ARM64_T6000
- Python: 3.11.4
- Stable-Baselines3: 2.0.0
- PyTorch: 2.0.1
- GPU Enabled: False
- Numpy: 1.25.1
- Cloudpickle: 2.2.1
- Gymnasium: 0.28.1
- OpenAI Gym: 0.26.2

无法确定卡顿原因,请求技术帮助。

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

相关产品推荐
方舟 Agent Plan

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

最近更新时间:2026.07.15 18:58:19