ubuntu从零开始安装mxnet--安装cuda8.0

CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
根据mxnet官网的说法,我们应该安装的是cuda8.0的版本。

下载cuda

这里笔者给出了最新版cuda的下载地址,https://developer.nvidia.com/cuda-downloads,8.0版本的下载地址不明原因无法打开。笔者给出了百度云的地址,不过只有Ubuntu16.04 cuda8.0.61版本。链接: https://pan.baidu.com/s/1nuCY4yd 密码: jcrj
如果小伙伴自行下载请选择.run文件进行下载,因为其他方式,笔者进行了多次尝试都很难成功。

驱动准备

安装nvidia显卡驱动,可以看一下笔者的上一篇文章《ubuntu从零开始安装mxnet–安装NVIDIA驱动》

运行cuda安装文件

  1. 安装辅助的库
    sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
  2. 运行文件
    ./cuda_8.0.61_375.26_linux.run
  3. 运行流程
    • 运行一开始会让我们阅读一段非常长的协议,我们可以通过ctrl + c跳过这段。并在之后第一个问题输入accept同意以上协议。
    • 不要安装nvidia驱动!!!!(关键),在问题中输入n
      这里写图片描述
    • 接下来我们只需要输入yes和保持默认路径即可
    • 安装成功
      这里写图片描述

检查和补充安装

  • 执行nvcc -V会出现两种情况。
    1. 第一种是我们之前的安装没有完全安装完成,会有如下提示
      这里写图片描述
      • 我们只需要根据提示执行sudo apt install nvidia-cuda-toolkit即可,不过可能时间有点久,请耐心等待。
    2. 经过一番耐心等待之后,安装完成,再次输入nvcc -V,得到如下结果
      这里写图片描述

运行一个例子

在安装完成后,我们可以尝试运行官方给的一个sample,如果我们之前安装的时候按照默认路径安装的话,不出意外会在/root/NVIDIA_CUDA-8.0_Samples文件夹下。
1. 执行cd /root/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery进入文件夹
2. 执行make编译文件
3. 执行./deviceQuery,如果出现了Result = PASS字样,恭喜,之前的步骤全部成功,可以进行下一步。
这里写图片描述

添加环境变量

编辑vim /etc/profile,在最后面添加

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

最后执行source /etc/profile

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章