ubuntu 16.04 下cuda的安装

前端之家收集整理的这篇文章主要介绍了ubuntu 16.04 下cuda的安装前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
  • cuda:CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。
  • cuDNN:CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案,目前支持的DL库包括Caffe,ConvNet,Torch7等。

  • cuda 的安装

    • cuda下载官网:https://developer.nvidia.com/cuda-downloads
    • 按照如下的配置下载
    • 下载完成后再执行下述操作

      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
  • 检测cuda是否安装成功
    • 配置环境变量

      sudo vim ~/.bash_profile
      # 加入
      export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
      export CUDA_HOME=/usr/local/cuda
      # 保存后执行
      source ./bash_profile
    • 执行nvidia-smi,如果出现下述界面则安装成功
    • 遇到的问题
      • Failed to initialize NVML: Driver/library version mismatch:重启就可以解决
  • CuDNN 的安装
    • 进入网址,注册登录后方能下载 https://developer.nvidia.com/cudnn
    • 按照下图所示,进行操作
    • 下载完成后执行下述操作:
      tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz 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*

猜你在找的Ubuntu相关文章