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安装文件
- 安装辅助的库
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
- 运行文件
./cuda_8.0.61_375.26_linux.run
- 运行流程
- 运行一开始会让我们阅读一段非常长的协议,我们可以通过ctrl + c跳过这段。并在之后第一个问题输入accept同意以上协议。
- 不要安装nvidia驱动!!!!(关键),在问题中输入n
- 接下来我们只需要输入yes和保持默认路径即可
- 安装成功
检查和补充安装
- 执行
nvcc -V
会出现两种情况。
- 第一种是我们之前的安装没有完全安装完成,会有如下提示
- 我们只需要根据提示执行
sudo apt install nvidia-cuda-toolkit
即可,不过可能时间有点久,请耐心等待。
- 我们只需要根据提示执行
- 经过一番耐心等待之后,安装完成,再次输入
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