Ubuntu16.04+cuda9.0安装教程

Ubuntu16.04+cuda9.0安装教程

转载自：https://www.cnblogs.com/iloveblog/p/7683349.html

1.安装NVIDIA驱动

首先去官网（http://www.nvidia.cn/Download/index.aspx?lang=cn）查找适配自己电脑GPU的驱动，
　　

安装驱动见上一篇博客。装好后输入nvidia-smi，输出gpu信息证明安装成功。

2.安装CUDA9.0

首先去官网 https://developer.nvidia.com/cuda-toolkit-archive
　　下载CUDA 9.0，下载的是.run文件，下载for Linux 版本。

下载完CUDA 9.0之后执行如下语句，运行.run文件

sudo sh cuda_9.0.176_384.81_linux.run

单击回车，一路往下运行，直到提示“是否为NVIDIA安装驱动nvidia-384？”，选择否，因为已经安装好驱动程序了，其他的全都是默认，不过要记住安装位置，默认是安装在/usr/local/cuda文件夹下。

配置环境变量，运行如下命令打开profile文件

sudo gedit  /etc/profile

打开文件后在文件末尾添加路径，也就是安装目录，命令如下：

export  PATH=/usr/local/cuda-9.0/bin:$PATH

export  LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH　

保存，然后重启电脑

sudo reboot

3.测试CUDA的Samples例子

　cd  /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
　sudo make -j4     #这是多线程，可以根据自己电脑配置选，也可以不要-j4，那就是单线程

　./deviceQuery

如果显示的是关于GPU的信息，则说明安装成功了。

4.查看cuda和cudnn版本

查看cuda版本：

cat  /usr/local/cuda/version.txt

查看cudnn版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2