1. CUDA
下载安装
官网 https://developer.nvidia.com/cuda-downloads 下载cuda8.0安装文件 .run
sudo sh cuda_8.0.27_linux.run
(实际测试发现需要使用chmod +x 加可执行权限 然后./运行)
注意:执行后会有一系列提示让你确认,但是有个让你选择是否安装NVIDIA驱动时,如果你已经安装过NVIDIA驱动,一定要选择否:
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 361.62?
因为前面我们已经安装了NVIDIA驱动,所以这里不要选择安装。其余的都直接默认或者选择是即可。
环境变量配置
打开~/.bashrc文件:
sudo vim ~/.bashrc
将以下内容写入到~/.bashrc尾部:
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
测试CUDA的samples
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery #由自己电脑目录决定
make
sudo ./deviceQuery
nvcc -V #or nvcc --version
如果显示一些关于GPU的信息,则说明安装成功。
2. cuDNN
下载cuDNN
需要注册NVIDIA开发者账号。注意cuDNN的版本需要和CUDA版本对应。注册后的下载地址https://developer.nvidia.com/rdp/cudnn-download
$ cd folder/extracted/contents
$ sudo cp -P include/cudnn.h /usr/include
$ sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
$ sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*
Refer
[1]. https://askubuntu.com/questions/799184/how-can-i-install-cuda-on-ubuntu-16-04
[2]. https://askubuntu.com/questions/767269/how-can-i-install-cudnn-on-ubuntu-16-04?rq=1
[3]. http://www.linuxidc.com/Linux/2017-01/139313.htm