专栏首页机器学习、深度学习Ubuntu16.0.4+1080ti+ cuda9.1+cudnn7.0.5 安装 及 caffe 测试

Ubuntu16.0.4+1080ti+ cuda9.1+cudnn7.0.5 安装 及 caffe 测试

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhangjunhit/article/details/79402781

主要还是要参考官方文档 cuda 安装有两种方法,这里参考 blog.csdn.net/guojunxiu/article/details/78848503 使用 Package Manager Installation的方法

目前网上大部分教程都是Runfile Installation安装的,然而这中方法很繁琐,需要关X Server,禁用nouveau等等,还会出现无法开机、无限循环登录等一些列毁天灭地的后果。如果侥幸顺利安装后,一个”CUDA driver version is insufficient for CUDA runtime version”的错误可以让你玩一年~ 装了N次之后我已经有些绝望了,后来发现还有一种Package Manager Installation的方法,只需要四行命令就解决了,想起之前真的是天堂有路你不走,地狱无门你自投o(╥﹏╥)o

—————————————吐槽分割线—————————————————— 1. Pre-installation Actions

这一步需要检查

GPU是否支持CUDA
Ubuntu版本是否受支持
gcc版本检查
Kernel Headers and Development Packages 是否已安装

这些步骤官方教程很详细,请移步官方教程第二节Pre-installation Actions 2. 下载Nividia Cuda Toolkit的deb安装包

CUDA Toolkit下载地址 根据自己的系统一步一步选择,注意主要下载的是.deb文件 3. 执行命令进行安装

在deb所在目录下依次执行以下命令即可。 1. sudo dpkg -i cuda-repo-ubuntu1604-9-1-local_9.1.85-1_amd64.deb 2. sudo apt-key add /var/cuda-repo-<\version>/7fa2af80.pub # 这句命令有提示 3. sudo apt-get update 4. sudo apt-get install cuda 以上命令其实就是选择完deb安装包后,官方给出的安装说明。 deb安装包会安装CUDA Toolkit和Driver Package,不需要自己安装驱动 安装时好像也会自动设置环境变量,记不太清了~如果没有的话请查看官方教程7.1.1. Environment Setup,设置一下环境变量。 /// 这里是需要设置一下的,具体参考官方文档

  1. 验证安装

重启系统然后进行验证 1. 检查GPU驱动版本和NVCC版本 2. 编译NVIDIA_CUDA-9.1_Samples下的例子,然后跑两个可执行文件。

官方文档也很清楚,请移步官方简称的7.2.3节Verify the Installation

cudnn 安装: 2.3.1. Installing from a Tar File 1. Navigate to your directory containing the cuDNN Tar file. 2. Unzip the cuDNN package. $ tar -xzvf cudnn-9.0-linux-x64-v7.tgz 3. Copy the following files into the CUDA Toolkit directory. $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 $ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

2.3.2. Installing from a Debian File 1. Navigate to your directory containing cuDNN Debian file. 2. Install the runtime library, for example: 具体名称以下载的 deb文件名为准 sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb 3. Install the developer library, for example: sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb 4. Install the code samples and the cuDNN Library User Guide, for example: sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.

caffe编译的问题解决:“cublas_v2.h: No such file or directory”

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include /usr/local/cuda-9.1/targets/x86_64-linux/include

LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial /usr/local/cuda-9.1/targets/x86_64-linux/lib

#

GTX1080 LetNet-5 CPU GPU cuDNN6.0 时间对比

CPU模式: Makefile.config

CPU_ONLY:=1 前取消 # make clean make -j

./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt Testing for 50 iterations

Total Time: 2906 ms.

GPU 模式:

Makefile.config

#CPU_ONLY:=1 make clean make -j

./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt -gpu 0

Total Time: 370.198 ms.

cuDNN 模式:

Makefile.config

USE_CUDNN:=1

make clean make -j

./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt -gpu 0

Testing for 50 iterations. Total Time: 68.2353 ms.

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 图像拼接--Seam Carving for Content-Aware Image Resizing

    Seam Carving for Content-Aware Image Resizing 智能图像缩放 ACM Transactions on graph...

    用户1148525
  • 模型量化

    TBN:Convolutional Neural Network with Ternary Inputs and Binary Weights ECCV_20...

    用户1148525
  • 统计学习导论 Chapter3--Linear Regression

    Book: An Introduction to Statistical Learning with Applications in R http:...

    用户1148525
  • Shiro 修改权限,刷新权限

    结合缓存框架,第一次查询权限信息后缓存,在缓存生存期内,访问鉴权使用缓存中的权限信息。

    WindWant
  • 【jQuery进阶】子菜单插件Slight Submenu

    兼容所有浏览器(记住,jQuery的2 *及以上不支持<IE9,如果您使用的是,对于那些旧的浏览器不支持)

    用户5640963
  • HDU 1013 Digital Roots【字符串,水】

    Digital Roots Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/3276...

    Angel_Kitty
  • Foreground-Constrained Eulerian Video Magnification

    HaHack
  • 详述 Elasticsearch 集群的搭建步骤及方法

    版权声明:Follow your heart and intuition. https://...

    CG国斌
  • 在线matlab代码学习神器Octave Online

    Octave与MATLAB完全兼容,免安装使用方便。注册和非注册功能会有不同,如下:

    zhangrelay
  • 在线matlab代码学习神器Octave Online

    Octave与MATLAB完全兼容,免安装使用方便。注册和非注册功能会有不同,如下:

    zhangrelay

扫码关注云+社区

领取腾讯云代金券