ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[FEATURE]: Optimization

Open programmin1 opened this issue 2 years ago • 2 comments

Describe the feature

You mention CUDA optimization which is "needed" but why do you not have support for Movidius compute stick or a Pine64 Quartz64 Neural Network Acceleration Engine? Is there a cpu strong enough at working this that it can be done without Nvidia-CUDA? Can that be supported as well or does it provide enough power?

programmin1 avatar Feb 20 '23 00:02 programmin1

Interesting feature, however our primary target is the training process which occurs mostly on GPU platform. However, edges devices can be an important scenario for inference, we probably will look into this feature in the future.

FrankLeeeee avatar Feb 21 '23 07:02 FrankLeeeee

In other words it will not train OR run without CUDA?

programmin1 avatar Feb 22 '23 03:02 programmin1

Hi @programmin1 Yes, you can check the detailed requirements. https://github.com/hpcaitech/ColossalAI#Installation This issue was closed due to inactivity. Thanks.

binmakeswell avatar Apr 20 '23 04:04 binmakeswell