NeMo-Curator
NeMo-Curator copied to clipboard
[FEA] Remove GPU-related messages on CPU-only servers
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