首先验证你是否有nvidia的显卡(http://developer.nvidia.com/cuda-gpus这个网站查看你是否有支持gpu的显卡):
$ lspci | grep -i nvidia
查看你的linux发行版本(主要是看是64位还是32位的):
$ uname -m && cat /etc/*release
看一下gcc的版本:
$ gcc --version
首先下载nvidia cuda的仓库安装包(我的是ubuntu 14.04 64位,所以下载的是ubuntu14.04的安装包,如果你是32位的可以参看具体的地址,具体的地址是https://developer.nvidia.com/cuda-downloads)
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_6.5-14_amd64.deb
下载完成之后可以使用如下命令安装它,注意文件名修改为cuda-repo-ubuntu1404_6.5-14_amd64.deb
sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb
安装好仓库之后,就可以更新你的本地仓库。
sudo apt-get update
(我遇到的问题:http://quabr.com/39322709/getting-cuda-8-0-to-work-on-ubuntu-16-04-nvidia-driver-issue)
最后开始安装cuda以及显卡驱动(安装cuda的同时就会把显卡驱动也全部安装好,这个真的很方便。但是下载的时间有点长。)
sudo apt-get install cuda
安装完之后你需要设置环境变量:
$ export PATH=/usr/local/cuda-6.5/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-6.5/lib64:$LD_LIBRARY_PATH
设置完毕之后,你还可以选择是否安装cuda附带的示例代码(<dir>
表示你要安装的位置,你可以将<dir>
替换成~
):
$ cuda-install-samples-6.5.sh <dir>
接下来做一些验证工作:
查看显卡的驱动版本
cat /proc/driver/nvidia/version
查看nvcc编译器的版本
nvcc -V i
编译cuda的示例代码:
cd ~/NVIDIA_CUDA-6.5_Samples
然后make一下编译代码:进入bin路径运行devicequery
cd ~/NVIDIA_CUDA-6.5_Samples/bin
./ deviceQuery