Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Running RAPIDS without GPU for development?

Tags:

gpu

rapids

cudf

Is there a way to run RAPIDS without a GPU? I usually develop on a small local machine without a GPU, then push my code to a powerful remote server for real use. Things like TensorFlow allow switching between the CPU and GPU depending on if they're available. Can an equivalent thing be done with RAPIDS? Even if it's slow, being able to test things on a machine without a GPU would be extremely helpful.

like image 526
golmschenk Avatar asked Sep 03 '25 14:09

golmschenk


1 Answers

There isn't a way to use RAPIDS without a GPU, and part of the reason for that is we're following the APIs the community has adopted in CPU packages across Pandas, Numpy, SKLearn, NetworkX, etc. This way it should be as easy as swapping an import statement to get something working on the CPU vs the GPU.

like image 163
Keith Kraus Avatar answered Sep 05 '25 15:09

Keith Kraus