Workaround to avoid cupy build error

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

5 stars 3 forks source link

Workaround to avoid cupy build error #45

Closed rraminen closed 2 years ago

rraminen commented 2 years ago

Workaround to avoid "ERROR: Package 'cupy' requires a different Python: 3.6.13 not in '>=3.7'" during cupy build. This error started occurring after the latest IFU on https://github.com/ROCmSoftwarePlatform/cupy.git.

NOTE: This PR has to be reverted once ROCm switches to python 3.7 or above

cc: @jithunnair-amd @sunway513 @jeffdaily @amathews-amd