To test GPU support for Tensorflow, we should run the following according to the manual of Tensorflow
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
However, in my case, I saw an annoying message:
2023-02-07 14:40:01.345350: I tensorflow/compiler/xla/stream_executor/cuda/cuda_gpu_executor.cc:981] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
A Stack Overflow discussion has an excellent explanation about this. I have a single CPU and a single GPU installed on the system. The system is a Ubuntu 20.04 LTS. Following the advice given over there, the following command gets rid of the message,
su -c "echo 0 | tee /sys/module/nvidia/drivers/pci:nvidia/*/numa_node"
That is sweet!
No comments:
Post a Comment