跨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
相关产品推荐
相关产品推荐

