Linux 配置GPU环境

简单来说有三步：
1、安装Nvidia 驱动，
使用lspci |grep VGA或lspci |grep -i nvidia查看显卡信息
根据显卡型号在这里下载驱动：https://www.nvidia.com/Download/index.aspx?lang=en-us
或 https://www.geforce.com/drivers
2、安装CUDA Tookit，并设置环境变量
3、安装cuDNN， 参考：https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.2.1/prod/9.0_20180806/cudnn-9.0-linux-x64-v7.2.1.38

完整的步骤（可能有变化，具体参考Tensorflow文档： Tensorflow GPU on Ubuntu
）：

image.png

CUDA Tookit安装文档：https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions

CUDA Tookit下载地址:CUDA Toolkit 9.0 Downloads

cuDNN下载地址(gmail登陆会比较慢,用其它邮箱)：https://developer.nvidia.com/rdp/cudnn-download

cuDNN的版本不对会导致在使用tensorflow-gpu时报段错误。