search

getkeops / keops

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
https://www.kernel-operations.io
MIT License
1.04k stars 64 forks source link

How can I use cpu only for the keops? #176

Closed YifanDengWHU closed 3 years ago

YifanDengWHU commented 3 years ago

I have spent a lot of time dealing with the CUDA support problem. However, I don't have much experience and enough permission to deal with nvcc, cmake. So I decided to use CPU only to run through the data pipeline of a specific project. It said that I only need to set tagHostDevice=0, so where can I change this arrtibute? Thanks!

hypnopump commented 3 years ago

This is also discussed here: https://github.com/getkeops/keops/issues/170

The simplest answer would be: In the same session in which you will run the scripts, run export CUDA_VISIBLE_DEVICES= before, the run the pykeops script.

Hope this helps! Eric Alcaide

YifanDengWHU commented 3 years ago

Thanks!