美文网首页
ubuntu 16.04 安装 cuda 8 toolkit 及

ubuntu 16.04 安装 cuda 8 toolkit 及

作者: vola_lei | 来源:发表于2017-04-07 13:01 被阅读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 说明成功啦!

    相关文章

      网友评论

          本文标题:ubuntu 16.04 安装 cuda 8 toolkit 及

          本文链接:https://www.haomeiwen.com/subject/nykhattx.html