search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.84k stars 178 forks source link

Installing FFCV on CPU-only node #359

Open mengwanguc opened 9 months ago

mengwanguc commented 9 months ago

Hi,

I know that installing FFCV on CPU-only node is not officially supported at this point.

However, I'm wondering there is anything in FFCV that would conceptually prevent us using FFCV on CPU-only nodes? (e.g. some optimization/functionality code that is deeply coupled with CUDA/GPU to compile). (e.g. does FFCV offload some data preprocessing on GPU? If so, the compilation of that part of code might potentially be a problem.)

Or do you think it is conceptually doable, but would take time to figure out the correct environments?

Thanks! Meng

mengwanguc commented 9 months ago

Any thought will be appreciated, thanks! @GuillaumeLeclerc @andrewilyas