DeepSpeed 是一个用于训练超大规模模型的优化库,它的目标是提高 PyTorch 的训练性能和规模。DeepSpeed 可以与 PyTorch 1.x 版本一起使用,并且不支持 PyTorch 2.0 版本。目前(2021年7月),PyTorch 的最新稳定版本是 1.9.0,而 PyTorch 2.0 尚未发布,因此 DeepSpeed 目前不支持 PyTorch 2.0。当 PyTorch 2.0 正式发布后,DeepSpeed 可能会进行相应的更新以支持新版本的 PyTorch。

DeepSpeed 目前不支持 PyTorch 2.0 - 了解最新情况

原文地址: https://www.cveoy.top/t/topic/qdtQ 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录