在Windows 10上安装tensorflow-gpu 1.3.0

时间:2017-10-16 19:45:09

标签: tensorflow tensorflow-gpu

我一直试图通过

在Windows 10上安装tensorflow-gpu

pip3 install --upgrade tensorflow-gpu

当我这样做时,我打破了普通tensorflow!的当前安装,并​​收到此错误:On Windows, running "import tensorflow" generates No module named "_pywrap_tensorflow" error

不知怎的,我设法通过重新安装普通的tensorflow来解决这个问题,但是当我在python 3.5.2中导入tensorflow并尝试识别我的GPU时,找不到设备!

我在CUDA / v9.0 / bin中定义为cudnn64_6并安装了Cuda 9.0,我可以毫无问题地运行nbody测试程序,我可以看到GPU用于该演示应用程序。

tensorflow-gpu 1.3.0是否有任何已知问题?

真的在这方面苦苦挣扎。为什么安装这个库时必须如此有问题!

请帮忙 毫克

1 个答案:

答案 0 :(得分:0)

TensorFlow 1.3(和1.4)需要CUDA 8.0,不支持更高版本。您需要将CUDA降级到8.0或从源进行自定义构建。