Unity运行ML-Agents报错RuntimeError: numpy.float32 dtype推断失败
解决ML-Agents运行时"RuntimeError: Could not infer dtype of numpy.float32"报错
版本兼容性调整
你标注的Python版本大概率存在笔误(应为Python 3.10.0),ml-agents 0.30.0对Python 3.10的支持不完善,建议降级Python到3.8或3.9版本;或者升级ml-agents到0.31.0及以上,同时将PyTorch升级到1.12.0+cpu版本,确保各依赖版本匹配。指定numpy兼容版本
该报错多因numpy与PyTorch的dtype交互异常导致,安装与PyTorch 1.11.0兼容的numpy版本:pip install numpy==1.21.6显式指定张量dtype
若使用了自定义环境或模型代码,检查numpy数组转PyTorch张量的代码,显式声明dtype避免自动推断失败:# 替换原代码 torch.tensor(numpy_array) # 改为 torch.tensor(numpy_array, dtype=torch.float32)清理重装依赖
卸载现有冲突依赖后重新安装:# 卸载全部相关包 pip uninstall -y ml-agents ml-agents-envs torch numpy # 按顺序安装指定版本 pip install numpy==1.21.6 pip install torch==1.11.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html pip install ml-agents==0.30.0 ml-agents-envs==0.30.0
内容的提问来源于stack exchange,提问作者IwanFtw
相关产品推荐
相关产品推荐

