ubuntu16.04安装cuda

Posted on

环境

ubuntu16.04.01桌面

kernel 4.4.0

RTX270

禁用nouveau

nouveau是默认的显卡驱动，一般linux出厂都使用该驱动，要禁止它

创建文件:

vilink
1
2

添加内容

blacklist nouveau
options nouveau modeset=0

1
2

更新```initramfs

1
sudo update-initramfs -u

重启

| grep nouveau```
1
2
3
4
5
6
7
8

没有输出则禁用成功

#### 关闭X Server

`CTRL+ALT+F1`进入tty1

```sudo service lightdm stop

安装cuda

习惯采用run文件安装，连驱动一起安装，之前都是ubuntu server上，很容易就安装上了，但是换了桌面，驱动死活安装不上，总是报错

1
The driver installation is unable to locate the kernel source. Please make sure that the kernel source packages are installed and set up correctly. If you know that the kernel source packages are installed and set up correctly, you may pass the location of the kernel source with the '--kernel-source-path' flag

所以单独安装，先安装驱动

下载驱动：https://www.nvidia.com/Download/index.aspx?lang=en-us，选择适合自己的系统的选项，页面提示推荐的驱动版本，下载

安装驱动

1
sudo sh NVIDIA-Linux-x86_64-430.40.run

提示

distribution-provided pre-install script failded! Are you sure you want to continue?```，不用理会，选择```continue installation```
1
2
3
4
5
6
7
8
9
10
11
12

```warming```不理会，继续安装

```Would you like to run the nvidia-xconfig utility…```选择是，安装完毕，输入```nvidia-smi```可以显示驱动，显卡名称等信息

再安装cuda

下载：<https://developer.nvidia.com/cuda-92-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal>

安装：``sudo sh cuda_9.2.148_396.37_linux.run``,选择的时候驱动不要安装

设置环境变量

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

1
2

打开x-server

sudo service lightdm start

1
2
3
4



#### 测试cuda安装

sudo make -j4
cd path/NVIDAI_CUDA-9.2_Samples/bin/x86_64/linux/releases
./deviceQuery

1
2

以上过程必要是安装如下更新，我执行过下面这些步骤，但是不知道是否起作用

sudo apt-get update
sudo apt-get upgrade
sudo apt dist-upgrade

1
2
3
4
5
6



#### cudnn安装

下载对应的cudnn版本，解压后执行：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

1
2

查看cudnn版本：

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

参考

https://developer.download.nvidia.cn/compute/cuda/9.2/Prod2/docs/sidebar/CUDA_Installation_Guide_Linux.pdf

https://zhuanlan.zhihu.com/p/36979657

https://zhuanlan.zhihu.com/p/68069328