实测Pytorch和Horovod版本对应

不同的Pytorch需要安装不同版本的horovod,否则horovod安装失败的时候提示的错误信息你都不知道到底是什么原因。。

我自己实测了几个版本,记录如下:

Pytorch Horovd
1.7.1+cu101 0.19.5
1.8.1+cu102 0.22.1
1.9.0+cu111 0.23.0

后续会持续更新

posted @ 2021-11-03 12:31  marsggbo  阅读(721)  评论(0编辑  收藏  举报