升级Tensorflow至1.3.0和cuDNN

我在Ubuntu 16.04，cuDNN 5.1.5和CUDA升级我TensorFlow 1.0.1与升级Tensorflow至1.3.0和cuDNN

$ pip install --upgrade tensorflow-gpu

8.0

然而，有错误消息，当我输入tensorflow在Python模块如下：

>> import tensorflow as tf 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/__init__.py", line 24,           in <module> 
    from tensorflow.python import * 
    File "/usr/local/lib/python2.7/dist- 
packages/tensorflow/python/__init__.py", line 49, in <module> 
    from tensorflow.python import pywrap_tensorflow 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow.py", line 52, in <module> 
    raise ImportError(msg) 
ImportError: Traceback (most recent call last): 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow.py", line 41, in <module> 
    from tensorflow.python.pywrap_tensorflow_internal import * 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>_pywrap_tensorflow_internal = swig_import_helper() 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description) 
ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory 

Failed to load the native TensorFlow runtime.

我认为这是与cuDNN版本兼容性的问题。任何人都知道如何升级cuDNN并使其与tensorflow 1.3.0兼容？

来源

2017-09-02 WKIm

tensorflow == 1.3.0不支持'cudnn = 5.1'它需要'cudnn = 6' –

我这么认为。但我想知道如何升级cuDNN。我不知道安装cuDNN 6是否足够..不需要'configure'步骤？ – WKIm

TensorFlow 1.3需要cuDNN版本6.请按照NVIDIA's cuDNN website上的平台安装说明进行操作。一旦cuDNN 6安装成功，您的TensorFlow 1.3的安装应该开始工作。

来源

2017-09-04 23:52:22 mrry

我写了一个脚本，可以使用干净的安装来升级cuDNN版本。您可以从下载脚本：

https://github.com/dnzzcn/cuDNNv

这是脚本的作用：

#!/bin/bash 

rm -f /usr/include/cudnn.h 
rm -f /usr/lib/x86_64-linux-gnu/*libcudnn* 
rm -f /usr/local/cuda-*/lib64/*libcudnn* 


cp -P packages/cudnn/include/cudnn.h /usr/include 
cp -P packages/cudnn/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/ 
chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn* 

rm -rf packages/cudnn

它您需要的版本可以自动执行安装操作。

如果您有TensorFlow 1.3.0，它需要cuDNN v6.0，那么使用脚本安装cuDNN v6.0就足够了。

来源

2017-09-11 21:31:35 dnzzcn

升级Tensorflow至1.3.0和cuDNN

回答

相关问题