NeMo-Curator icon indicating copy to clipboard operation
NeMo-Curator copied to clipboard

[FEA] Remove GPU-related messages on CPU-only servers

Open miguelusque opened this issue 9 months ago • 1 comments
trafficstars

Hi! Would it be possible to hide this kind of errors when running NeMo Curator in a CPU only server?

/usr/local/lib/python3.10/dist-packages/cudf/utils/_ptxcompiler.py:61: UserWarning: Error getting driver and runtime versions:

I have used the following without success. Thanks!

export NUMBA_DISABLE_CUDA=1 export CUDF_PANDAS_FALLBACK_MODE=1

miguelusque avatar Feb 10 '25 15:02 miguelusque