【发布时间】:2020-01-14 10:06:00
【问题描述】:
这可能是重复的帖子,但其他帖子都没有为我工作。 我有 tensorflow 1.14.0 和 tensorflow-gpu 1.14.0 我已经安装了所需的兼容 Cuda 和 Cudnn 版本,并在环境变量中指定了它们的路径。但仍然是 tensorflow 和/或 tensorflow-gpu,无法识别我的 GPU。 所以验证码的:
from tensorflow.python.client import device_lib
def get_available_devices():
local_device_protos = device_lib.list_local_devices()
return [x.name for x in local_device_protos]
print(get_available_devices())
是:
['/device:CPU:0']
OP:Windows 10
IDE:Jupyter (anaconda)
GPU 版本:GTX 960M
注意:有些帖子提到你必须在最后卸载 tensorflow,但是当我这样做时,上面的代码甚至不起作用......
【问题讨论】:
-
请发布更多错误信息?你能运行 nvidia-smi 吗?
标签: python tensorflow gpu