NVIDIA GPU驱动安装

一、CUDA安装

./cuda_10.0.130_410.48_linux

安装后会有如下提示：

查看CUDA的版本号

二、CUDNN安装

安装命令

> tar xvzf cudnn-10.0-linux-x64-v7.5.0.56.tgz

> cp cuda/include/cudnn.h /usr/local/cuda/include/

> cp cuda/lib64/* /usr/local/cuda/lib64/

> ldconfig /usr/local/cuda

查看CUDNN版本

三、GPU驱动安装

./NVIDIA-Linux-x86_64-410.57.run

验证方法

执行nvidia-smi，如果一切正常会列出当前机器上的显卡。

如果发现nvidia-smi执行很慢，可以执行下nvidia-persistenced --persistence-mode，再执行nvidia-smi就会变快很多。