tensorflow-gpu无法成功安装

时间:2019-03-31 15:28:49

标签: python tensorflow gpu nvidia

我正在安装tensorflow-gpu-1.5和CUDA v.9和cuDNN v.7,但是我的tensorflow无法与gpu一起使用。当我如下检查时:

    from tensorflow.python.client import device_lib
    print(device_lib.list_local_devices())

返回:

    name: "/device:CPU:0"
    device_type: "CPU"

我很困惑如何使tensorflow与gpu一起使用。我可以逐步安装什么。我正在安装CUDA和cuDNN,是否需要更多要求? NVIDIA驱动程序呢?我正在使用Windows10,Python3.5,NVS 5200M。 谢谢。

0 个答案:

没有答案