2017-10-16 114 views
1

我一直在試圖安裝在Windows 10 tensorflow-GPU,通過安裝tensorflow-GPU在Windows 1.3.0 10

PIP3安裝--upgrade tensorflow-GPU

當我做到這一點我打破目前安裝普通的tensorflow !,並得到這個錯誤:On Windows, running "import tensorflow" generates No module named "_pywrap_tensorflow" error

不知何故,我設法通過重新安裝普通tensorflow來解決這個問題,但是當我在python 3.5.2中導入tensorflow並嘗試識別我的GPU時,找不到設備!

我有一個CUDA 9.0與cudnn64_6一起安裝,定義爲CUDA/v9.0/bin中的一個DLL,並且我可以運行nbody測試程序而沒有問題,我可以看到該GPU正在用於該演示應用程序。

tensorflow-gpu 1.3.0有沒有已知的問題?

真的在這掙扎。爲什麼安裝這個庫需要這麼麻煩!

請幫 毫克

回答

0

TensorFlow 1.3(1.4)要求CUDA 8.0,不支持更高版本。您需要將CUDA降級到8.0或從源代碼進行自定義構建。

+0

很多感謝您的幫助!它現在有效。 – mg0