我一直试图通过
在Windows 10上安装tensorflow-gpupip3 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是否有任何已知问题?
真的在这方面苦苦挣扎。为什么安装这个库时必须如此有问题!
请帮忙 毫克
答案 0 :(得分:0)
TensorFlow 1.3(和1.4)需要CUDA 8.0,不支持更高版本。您需要将CUDA降级到8.0或从源进行自定义构建。