首页 > 解决方案 > 在 Jupyter Notebook 中未检测到 GPU

问题描述

在 Jupyter VM 上,当我们执行 nvidia-smi 时,它在后端检测 GPU,但是当我们尝试使用命令运行 tensorflow 时,它没有显示在应用程序上。命令:

import tensorflow as tf
tf.test.is_gpu_available()

执行后tf.test.is_gpu_available(),尽管在后端检测到 GPU,但我们得到了错误。

如果有任何解决此问题的想法,请提供一些输入。

标签: tensorflow

解决方案


推荐阅读