search

ROCm / DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
https://www.deepspeed.ai/
Apache License 2.0
4 stars 3 forks source link

Make torch version check numeric #27

Closed rraminen closed 3 years ago

rraminen commented 3 years ago

This fix avoids DeepSpeed build errors on ROCm.