自己的主机是否用上了GPU,GPU的驱动是否正常,是否安装了正确的支持库;(后附完整代码)
python程序可以 检查 GPU驱动是否安装正确:
以下是 在腾讯云 虚拟主机的运行结果:
2023-03-16 13:58:55.888036: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
2023-03-16 13:58:57.108726: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcuda.so.1
2023-03-16 13:58:57.121681: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:937] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero<