cuda caffe cudnn

2024-08-27 20:18
文章标签 caffe cuda cudnn

本文主要是介绍cuda caffe cudnn,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

本系列文章由 @yhl_leo 出品,转载请注明出处。
文章链接: http://blog.csdn.net/yhl_leo/article/details/50961542

花了一天时间,在电脑上安装配置了Caffe深度学习框架,网上的很多教程和指导都已经过期,中间辗转耗费了点时间,这里把个人认为最简单的方法整理如下。

1 版本

  • 笔记本:ThinkPad W541
  • Ubuntu 14.04(64-bit)
  • 双显卡:
    • Intel(R) HD Graphics Family
    • NVIDIA Quadro K2100M
  • CUDA 7.5
  • cuDNN

2 准备工作

系统原本是想安装版本Wily,之前也一直在使用,但是无奈,电脑无法安装该版本,就选用了14.04。

安装一些基本依赖项:

$ sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler  
$ sudo apt-get install --no-install-recommends libboost-all-dev
$ sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
$ sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

安装NVIDIA显卡驱动,我是先卸载已有版本,再安装,卸载方法:

sudo apt-get remove --purge nvidia*

重启后登录时,可能会循环出现填写登录密码,不能进入系统的情况,按Ctrl+Alt+F1,登录,安装显卡驱动:

$ sudo apt-get install nvidia-current  

或者指定版本:

$ sudo add-apt-repository ppa:xorg-edgers/ppa
$ sudo apt-get update
$ sudo apt-get install nvidia-352

电脑是集成显卡(NVIDIA独显的不需要这么做),网友建议在安装前禁掉一些服务,以保证成功。我在安装的时候,只禁掉一项:

$ sudo service lightdm stop

(系统启动后,不要登录进入桌面,直接按Ctrl+Alt+F1,如果你玩过云服务器,应该很熟悉,接着输入系统用户名和登录密码,输入上述的指令,回到图形界面的方法是按Ctrl+Alt+F7,记住这个操作,14.04版本出现输入密码不能进入桌面的情况比较常见,我安装搜狗拼音都出现了……)

3 CUDA

网上很多安装6.x,7.0版本的方法非常繁琐,个人选用了相对简单的安装方法。

首先在官网上下载安装文件(链接前文已经提供):

cuda

我下载的是deb本地安装文件,下载完成后,按照文档提示的命令安装:

$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

下载了下方的Installation Guide for Linux,里面有关于环境变量的设置方法:

$ export PATH=/usr/local/cuda-7.5/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

其他安装文件的配置方式,文档里也都有。

4 Caffe

下载caffe:

$ git clone https://github.com/BVLC/caffe.git

如果你没安装Git,请阅读博客:Ubuntu Git安装与使用。

然后编译caffe:

$ cd ~/caffe
$ sudo cp Makefile.config.example Makefile.config
$ make all

有些教程也配置OpenCV,方法可以阅读博客: Ubuntu 配置OpenCV 。

配置OpenCV,请阅读:Ubuntu 15.10 OpenCV 3.1 Installation Guide(原贴已删)

配置OpenCV,请阅读:OpenCV 3.1 Installation in Linux。

5 cuDNN

cuDNN是GPU加速计算深层神经网络的库(下载链接,前文已提供)。

cuDNN

本人的下载文件是:cudnn-7.0-linux-x64-v4.0-prod.tgz

在终端中切换到文件所在文件夹,输入下面指令:

$ sudo tar xvf cudnn-7.0-linux-x64-v4.0-prod.tgz
$ cd cuda/include
$ sudo cp *.h /usr/local/include/
$ cd ../lib64
$ sudo cp lib* /usr/local/lib/
$ cd /usr/local/lib
$ sudo chmod +r libcudnn.so.4.0.4
$ sudo ln -sf libcudnn.so.4.0.4 libcudnn.so.4
$ sudo ln -sf libcudnn.so.4 libcudnn.so
$ sudo ldconfig

(我安装的是libcudnn.so.4.0.7,跟上面的例子对应就好)
然后切换到caffe根目录下,将Makefile.config中的USE_CUDNN行前的#去掉:

$ cd ~/caffe
$ sudo vi Makefile.config

config

保存后重新编译:

$ sudo make clean
$ sudo make all
$ sudo make test  
$ sudo make runtest 

本人编译的时候最后一步的时候,出现这样的错误:

libcudart.so.7.5 cannot open shared object file: No such file or directory

网上有一堆介绍,怎么设置环境变量的(我们分明已经设置过了),可以这样检查是否已经添加环境变量:

$ echo $PATH
$ echo $LD_LIBRARY_PATH

path

可以看到,环境变量已经添加好。

解决方法是这样,将一些文件复制到/usr/local/lib文件夹下:

$ sudo cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcublas.so.7.5 /usr/local/lib/libcublas.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcurand.so.7.5 /usr/local/lib/libcurand.so.7.5 && sudo ldconfig

再次,尝试sudo make runtest命令,出现如下:

maketest

至此,caffe安装完成。


源自:个人博客网站:yhl’s blog

这篇关于cuda caffe cudnn的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1112668

相关文章

PyInstaller问题解决 onnxruntime-gpu 使用GPU和CUDA加速模型推理

前言 在模型推理时,需要使用GPU加速,相关的CUDA和CUDNN安装好后,通过onnxruntime-gpu实现。 直接运行python程序是正常使用GPU的,如果使用PyInstaller将.py文件打包为.exe,发现只能使用CPU推理了。 本文分析这个问题和提供解决方案,供大家参考。 问题分析——找不到ONNX Runtime GPU 动态库 首先直接运行python程序

使用openpose caffe源码框架训练车辆模型常见错误及解决办法

错误1:what():  Error: mSources.size() != mProbabilities.size() at 51, OPDataLayer, src/caffe/openpose/layers/oPDataLayer.cpp 原因:这是因为在网络模型中数据源sources和probabilities设置的参数个数不一样导致的,一个数据源对应一个概率 解决方法:只需要将网络文

caffe训练openpose相关资源

CPMTransformationParameter参数解析: https://www.jianshu.com/p/063a2159f0f2 genLMDB.py: https://www.jianshu.com/p/1cae32cbd36d OpenPose 参数说明: https://blog.csdn.net/zziahgf/article/details/84668319 openp

在Ubuntu16环境下用QT调试caffe

用qt调试caffe需要配置一些东西具体方法可以参考一下链接: https://www.zhihu.com/question/27987666/answer/80728225 https://blog.csdn.net/xg123321123/article/details/52817658

ubuntu16.04 caffe python3.5 安装

在ubuntu16.04 下 caffe安装python3.5可以参考 https://blog.csdn.net/tingtie1438/article/details/82085199 但这里面有一些需要补充和修正其中的一些东西 一、其中“二. 安装依赖项”里面是这样的: sudo apt-get install python3-pipsudo apt-get install pytho

解决caffe 编译过程中protobuf版本冲突的问题

在编译caffe python3版本时一直会出现如下错误,(安装caffe python3具体方法可参考:https://blog.csdn.net/tingtie1438/article/details/82085199 ): 通过其错误信息可知是protobuf出了问题,现在网上教程一般都是默认安装的 libprotobuf-dev 和 protobuf-compiler,对于pytho

ubuntu16.04 caffe(github源码cpu)+python3.5+opencv3.4.5安装编译

https://www.cnblogs.com/hanjianjian90/p/10604926.html

CUDA:用并行计算的方法对图像进行直方图均衡处理

(一)目的 将所学算法运用于图像处理中。 (二)内容 用并行计算的方法对图像进行直方图均衡处理。 要求: 利用直方图均衡算法处理lena_salt图像 版本1:CPU实现 版本2:GPU实现  实验步骤一 软件设计分析: 数据类型: 根据实验要求,本实验的数据类型为一个256*256*8的整型矩阵,其中元素的值为256*256个0-255的灰度值。 存储方式: 图像在内存中

ffmpeg安装测试(支持cuda支持SRT)

文章目录 背景安装ffmpeg直接下载可执行文件选择版本选择对应系统版本下载测试Linux下安装 查看支持协议以及编码格式 常见错误缺少 libmvec.so.1LD_LIBRARY_PATH 错误 GPU加速测试SRT服务器搭建下载srs5.0源码解压安装配置启动 SRT推流测试SRT播放测试 背景 在音视频开发测试中,FFmpeg是一个不可或缺的工具,它以其强大的音视频处理