不同的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 |
后續會持續更新
不同的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 |
后續會持續更新
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。