stable_baselines3对应的pytorch版本
时间: 2023-06-16 21:05:33 浏览: 822
Python库 | stable_baselines3-0.9.0a2.tar.gz
5星 · 资源好评率100%
stable_baselines3是基于PyTorch和TensorFlow 2.0开发的深度强化学习库,主要使用PyTorch实现。根据stable_baselines3的官方文档,它支持的PyTorch版本是1.7.0及以上。需要注意的是,stable_baselines3使用的是PyTorch的CPU版本,如果需要使用GPU加速,需要自行安装CUDA并安装对应的PyTorch GPU版本。
阅读全文