facebookresearch / deepcluster

Deep Clustering for Unsupervised Learning of Visual Features
Other
1.69k stars 325 forks source link

CPU or other CUDA versions supported? #51

Closed khg2478 closed 4 years ago

khg2478 commented 4 years ago

Hi. Due to hardware dependency, I cannot really use this library. I tried to set caffe in CPU mode and tried to run it but I face other issues.

Is there any other ways to run in CPU or other CUDA compatible modes?

Thanks.

mathildecaron31 commented 4 years ago

Hi, thanks for your interest. This library uses PyTorch, not Caffe. We do not provide CPU mode only unfortunately. Sorry for the inconvenience