ubuntu 16.04 安装 cuda 8 toolkit 及 cudnn 5.0

我的简书主页在这里。

安装 cuda 8.0

在ubuntu16.04 安装caffe依赖于cuda 8, 因此需要安装cuda 8 。

  1. 安装显卡驱动,直接在: system settings→software & updates→additional drivers 安装就行可以了。
  2. 安装 cuda toolkit 需要在命令行界面,首先需要关闭图形界面:
    sudo service lightdm stop

  3. 在终端运行指令 sudo sh cuda_8.0.27_linux.run
    不需要安装显卡驱动,这个选no其他都是yes.

  4. 安装完成后,需要装一些依赖库:
      sudo apt-get install freeglut3-dev build-essential libx11-dev
        libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa
        libglu1-mesa-dev

  5. 设置环境变量:
    在终端输入这两句

     export PATH=/usr/local/cuda-8.0/bin:$PATH
     Export LD_LIBRARY _PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
     

      然后在 文件中 sudo vi /etc/profile 输入上面这两句话,保存退出。
    最后输入:sudo ldconfig 使环境变量生效。

  6. 验证是否完成:
    nvidia-smi
    nvcc-V
    都会显示出相关信息。

安装 cudnn 5.0

安装 cuDNN 比较简单,解压后把相应的文件拷贝到对应的 CUDA 目录下即可: 1. 解压:tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz 文件夹中有一下内容: **cuda/include/cudnn.h cuda/lib64/libcudnn.so cuda/lib64/libcudnn.so.5 cuda/lib64/libcudnn.so.5.0.5 cuda/lib64/libcudnn_static.a* 2. 拷贝到相应目录,行以下指令: 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 sudo chmod a+r /usr/local/cuda/lib64/libcudnn* 3. 验证一下 cuda 的 samples 里面有个 deviceQuery 运行之后会显示信息,最后一行 出行 pass 说明成功啦!

相关文章

文章浏览阅读2.3k次,点赞4次,收藏22次。最近安装了CARLA预...
文章浏览阅读6.3k次,点赞5次,收藏15次。在清华镜像中下载U...
文章浏览阅读5k次。linux环境, python3.7.问题描述: 安装...
文章浏览阅读4.2k次,点赞4次,收藏17次。要安装这个 standa...
文章浏览阅读894次,点赞51次,收藏31次。在安卓使用vscode主...