ubuntu安装CUDA
因为深度学习需要用到CUDA,所以写篇博客,记录下自己安装CUDA 的过程。
1 安装前的检查
安装CUDA之前,首先要做一些事情,检查你的机器是否可以安装CUDA。
1.1 检查你的gpu是否是可以安装CUDA 的
运行如下命令:
$ lspci | grep -i nvidia
这个是我的机器的返回结果:
01:00.0 VGA compatible controller: NVIDIA Corporation GM107 [GeForce GTX 750 Ti] (rev a2)
01:00.1 Audio device: NVIDIA Corporation Device 0fbc (rev a1)
1.2 检查你的linux版本是否支持CUDA
运行如下命令:
uname -m && cat /etc/*release
我的机器返回结果如下:
x86_64
DISTRIB_ID=Ubuntu
DISTRIB_RELEASE=14.04
DISTRIB_CODENAME=trusty
DISTRIB_DESCRIPTION="Ubuntu 14.04.2 LTS"
NAME="Ubuntu"
VERSION="14.04.2 LTS, Trusty Tahr"
ID=ubuntu
ID_LIKE=debian
PRETTY_NAME="Ubuntu 14.04.2 LTS"
VERSION_ID="14.04"
HOME_URL="http://www.ubuntu.com/"
SUPPORT_URL="http://help.ubuntu.com/"
BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"
DISTRIB_ID=Ubuntu Kylin
DISTRIB_RELEASE=14.04
DISTRIB_CODENAME=trusty
DISTRIB_DESCRIPTION="Ubuntu Kylin 14.04"
x86_64代表你的机器是64位的,剩下的是解释的linux发行版信息。
如果是红帽,可能是这样的信息:
x86_64
Red Hat Enterprise Linux Workstation release 6.0 (Santiago)
CUDA只支持一些特定的linux发行版,有Fedora,OpenSuSE,RHEL,CentOS,SLES,Ubuntu.
1.3 验证操作系统是否安装了gcc
在使用CUDA Tookit 开发的时候,gcc是需要的,但是运行CUDA程序的时候不需要。
gcc -v
我的结果是:
Using built-in specs.
COLLECT_GCC=gcc
COLLECT_LTO_WRAPPER=/usr/lib/gcc/x86_64-linux-gnu/4.7/lto-wrapper
Target: x86_64-linux-gnu
Configured with: ../src/configure -v --with-pkgversion='Ubuntu/Linaro 4.7.3-12ubuntu1' --with-bugurl=file:///usr/share/doc/gcc-4.7/README.Bugs --enable-languages=c,c++,go