最近个人想用gpu版本的tensorflow-gpu v1.5(因为该版本支持cuda9.0和cudnn v7,所以就在unbuntu 16.04.05 LTS上安装了该cuda的版本),后来发现在python idle/ipython上是没有任何问题的,但在jupyter notebook上发现问题(因为我个人倾向用python pip来安装软件包,所以基本不用anconda的方式来安装,所以百度给出的大部分的解决方案都是基于anconda,希望能有遇到这些问题的一起探讨解决方案)。下面给出截图:
初步的解决方案:
经过自己的一番折腾与尝试,可能跟安装软件包的顺序(某种依赖)有关。后来把tensorflow-gpu v1.5.1、jupyter、ipykernel、cudn等都卸载了。重新安装。先安装cuda v9.0和cudnn v7.0,然后tensorflow-gpu v1.5.1,最后Jupyter和ipykernel等。