cuda_usage

查看本机所支持的 nvidia 驱动

ubuntu-drivers devices

recommend 可以直接用 apt 安装

nvidia 驱动官网下载:
https://www.nvidia.cn/Download/index.aspx?lang=cn#

安装cuda

在nvidia官网选择需要的cuda版本下载 https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal

安装方式选择runfile

安装后 cuda路径为/usr/local/cuda-XXXX

安装cudnn

在nvidia官网选择需要的cudnn版本下载 https://developer.nvidia.com/rdp/cudnn-archive

下载解压后，将文件夹内cuda/include/里的所有文件拷贝到/usr/local/cuda-10.2/include/文件夹内，将cuda/lib64/里的所有文件拷贝到/usr/local/cuda-10.2/lib64/文件夹内

查看gpu 硬件架构信息

https://blog.csdn.net/2301_77554343/article/details/134697452