安装显卡驱动，cuda,cudnn

安装步骤
环境：Ubuntu16.04

显卡：NVIDIA GTX970

安装显卡驱动
由于我们需要在Pytorch使用CUDA加速训练过程，因此第一步需要安装显卡驱动为安装CUDA做准备。

到NVIDIA官网下载相应的显卡驱动，地址是http://www.nvidia.com/Download/index.aspx?lang=en-usblacklist nouveau
options nouveau modeset=0
ctrl + alt + f1 #进入命令行模式以停止对显卡的使用
sudo service lightdm stop #关闭图形界面
sudo apt-get remove --purge nvidia* #删除旧的驱动
sudo apt-get update
sudo apt-get install dkms build-essential linux-headers-generic #安装依赖
sudo vim /etc/modprobe.d/blacklist-nouveau.conf #禁用nouveau，输入以下内容
保存上述文件后退出
sudo update-initramfs -u
重启电脑
ctrl + alt + f1 #再次进入命令行模式
sudo service lightdm stop #再次关闭图形界面
cd到驱动下载目录
sudo service lightdm stop
sudo ./NVIDIA-Linux-x86_64-384.98.run -no-opengl-files #安装驱动
sudo service lightdm start #启动图形界面
cat /proc/driver/nvidia/version #测试驱动
安装CUDA
到NVIDIA官网