lightning >=2.0.0
bagua-cuda116 >=0.9.2
