1.首先安装nvidia显卡驱动:
系统设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改
2.下载CUDA8.0地址https://developer.nvidia.com/cuda-release-candidate-download(需要登陆)
3.安装cuda
sudo dpkg -icuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
4.下载cudnn,最好是v4版本地址https:developer.nvidia.com/cudnn
5.把cudnn解压复制到cuda下
tar xvzf
cudnn-7.0-linux-x64-v4.0-prod.tgz
sudo cp cuda/include/cudnn.h/usr/local/cuda/include
sudo cp cuda/lib64/libcudnn*/usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h/usr/local/cuda/lib64/libcudnn*