stable_baselines3对应的pytorch版本
时间: 2023-06-16 07:05:33 浏览: 794
稳定基线3:稳定基线的PyTorch版本,强化学习算法的可靠实现
stable_baselines3是基于PyTorch和TensorFlow 2.0开发的深度强化学习库,主要使用PyTorch实现。根据stable_baselines3的官方文档,它支持的PyTorch版本是1.7.0及以上。需要注意的是,stable_baselines3使用的是PyTorch的CPU版本,如果需要使用GPU加速,需要自行安装CUDA并安装对应的PyTorch GPU版本。
阅读全文