Nvidia-smi 无法检测到 Gpu

Nvidia-smi 无法检测到 Gpu

我使用 tensorflow,它没有检测到 Gpu Nvidia-smi 正在显示这个,我该怎么办

在此处输入图片描述

pip 安装--用户 tensorflow-gpu

答案1

根据屏幕截图判断,您的 Nvidia 驱动程序已正确安装(因为使用 nvidia-smi,您能够看到驱动程序的具体信息,并且有一个可以访问 GPU 的正在运行的进程)。

但是,除了安装 Nvidia 驱动程序外,您还需要安装 CUDA-Toolkit 和 cuDNN 库。它们可供下载 -https://developer.nvidia.com/cuda-toolkithttps://developer.nvidia.com/cudnn

如果 CUDA 安装正确,那么你的 Tensorflow 安装可能与已安装的 CUDA 不兼容?请检查此处 -https://www.tensorflow.org/install/source#gpu

如果以上方法不适用,我建议验证您是否安装了支持 GPU 的 Tensorflow 版本 (Tensorflow-GPU)。您可以测试 Tensorflow 是否能够检测到您的 GPU -https://www.tensorflow.org/api_docs/python/tf/test/is_gpu_available

相关内容