zoukankan      html  css  js  c++  java
  • tensorflow 升级后报错:ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory

    我的tensorflow之前的版本是1.2的所以支持cudnn5,但是tensorflow1.3及以上就是支持cudnn6。

    查看:

    /usr/local/cuda/lib64$ ls  libcudnn*
    libcudnn.so  libcudnn.so.5  libcudnn.so.5.1.5  libcudnn_static.a

    因此要到官网下载cudnn6啦,官网链接:https://developer.nvidia.com/rdp/cudnn-download

    我下载的是:cudnn-8.0-linux-x64-v6.0.tgz :https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170307/cudnn-8.0-linux-x64-v6.0-tgz

    然后解压到本地,发现出现cuda文件夹,然后:

    1、覆盖之前cudnn5的头文件

    $ cd cuda/include 
    $ sudo cp *.h /usr/local/cuda/include/  (注意这里是你自己的cuda的安装地址,复制*.h文件) 

    2、覆盖lib库,记得把cudnn5删掉
    $ cd cuda/lib64  
    $ sudo rm /usr/local/cuda/lib64/libcudnn*
    $ sudo cp libcudnn* /usr/local/cuda/lib64/


    3(可选,好像不用下面这步骤也可以)
    $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 








  • 相关阅读:
    Java栈、堆、方法区
    Java冒泡算法
    JDK8 API下载
    java 对两个整数变量的值进行互换。三种方法
    Java中double相减精度的问题,和解决方法
    window.onload问题
    ES6中class详解
    YOLO V2
    YOLO V1
    ORB-SLAM2源码解析
  • 原文地址:https://www.cnblogs.com/YouXiangLiThon/p/8203401.html
Copyright © 2011-2022 走看看