1)安装Nvidia驱动
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-370
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
sudo reboot
2)下载cuda8.x的runfile安装包,安装CUDA8
chmod u+x ./cuda-8.0.44.linux.run
sudo ./cuda-8.0.44.linux.run --tmpdir=/tmp --override
接下来开始安装向导,除了是否安装自带的驱动,选择NO,其他的都是默认或者YES
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
(y)es/(n)o/(q)uit: n
安装完成后,添加环境变量
vi ~/.bashrc
把下面两行加到最后
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
3)测试
watch -n 0.1 nvidia-smi
测试cuda自带例子
cd ~/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery
make
./deviceQuery
4)安装cudnn
tar -zxvf cudnn-8.0-linux-x64-v5.0-ga.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
sudo ldconfig