Nvidia cuda cudnn安装

1. 删除之前驱动: sudo apt-get purge nvidia-*
    可能出现问题:
    无法获得锁 /var/lib/dpkg/lock - open (11: 资源暂时不可用) 
E: 无法锁定管理目录(/var/lib/dpkg/),是否有其他进程正占用它?”
     解决方法:
                 a. 终端:ps -aux   # 查看所有进程
                 b. 找到apt-get的进程的PID号,直接:sudo kill PID
                 c. sudo rm /var/cache/apt/archives/lock
     sudo rm /var/lib/dpkg/lock
2. 终端输入:ubuntu-drivers devices      # 查看适合电脑的驱动程序
       比如driver   : nvidia-387 - third-party free recommended    # 找到recommended推荐的版本
       sudo apt-get install nvidia-387         # 安装(较慢,可开启清华镜像)
3.  sudo reboot重启
输入nvidia-settings查看配置情况,nvidia驱动安装成功
4. 官网下载:
       cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
       cudnn-8.0-linux-x64-v5.0-ga.tgz
5. 下载目录下运行:(确保nvidia驱动安装成功再执行)
      sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb 
      sudo apt-get update 
      sudo apt-get install cuda
安装成功会在usr/local下看到cuda-8.0、cuda两个文件夹
6. 解压cudnn-8.0-linux-x64-v5.0-ga.tgz
cd cuda/include/ 
sudo cp cudnn.h /usr/local/cuda/include/   #复制cudnn.h到/usr/local/cuda/include下,注意这里是复制到cuda下不是cuda-8.0
cd ../lib64    
sudo cp lib* /usr/local/cuda/lib64/    # 同理复制lib*的三个文件
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

7.   cudnn解压后lib64下有libcudnn.so、libcudnn.so.5.0.5、libcudnn.so.5三个文件夹
       cd /usr/local/cuda/lib64/ 
      sudo rm -rf libcudnn.so libcudnn.so.5 
     sudo ln -s libcudnn.so.5.0.5 libcudnn.so.5 
    sudo ln -s libcudnn.so.5 libcudnn.so


8. 添加环境变量
     sudo gedit /etc/profile
    末尾加上:
     PATH=/usr/local/cuda/bin:$PATH 
    export PATH
    
    sudo vim /etc/ld.so.conf.d/cuda.conf
     按a进入插入模型输入:
        /usr/local/cuda/lib64
     然后esc  + :wq退出保存即可
     sudo ldconfig       # 生效
     
9.  sudo reboot      


10. 测试:
     /usr/local/cuda-8.0/sample下运行:sudo make
 再进入sample/1_Utilities/deviceQuery下:./deviceQuery
     最后输出Result = PASS即可

猜你喜欢

转载自blog.csdn.net/md2017/article/details/79039490