无法在GPU上运行Tensorflow

时间:2018-03-20 21:13:26

标签: ubuntu tensorflow nvidia cudnn

所以我在CPU上成功安装并运行了Tensorflow。事实证明,让它在GPU上运行非常困难。我的第一个问题是Cuda或其中一个GPU驱动程序附带的登录循环。在恢复模式下启动,清除nvidia驱动程序,最后添加最新的opensource Nvidia驱动程序(390)修复了该问题。此外,Cuda 9.1尚未得到支持。但安装旧的Cuda 9.0修复了这个问题。

现在,我收到错误:

  

加载的运行时CuDNN库:7101(兼容版本7100)但是   source使用7004编译(兼容版本7000)。

稍微阅读一下这似乎是一个将被修复的问题,但它还没有或者可能尚未稳定。

所以我尝试从这里下载一个较旧的deb包: https://developer.nvidia.com/rdp/cudnn-download
(另外7.0.0.5的唯一deb软件包适用于Ubuntu 16.04)

但问题仍然存在。它似乎没有寻找旧的包。

我唯一的选择是从源头建立吗?

(我在Ubuntu 17.04,GTX 970上)

0 个答案:

没有答案