安装教程主要针对 Linux (Ubuntu 20.04)

一、使用 Docker 镜像（推荐）

打包好的 cuda 镜像 https://hub.docker.com/r/nvidia/cuda https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda/tags cuda 镜像的 dockerfile 文件 https://gitlab.com/nvidia/container-images/cuda

Three flavors of images are provided:

cuda 镜像几个版本之间的关系为 base > runtime > devel > cudnn

推荐使用 vscode docker 插件，运行后就可直接在 vscode 中使用容器了

二、手动安装

1.显卡驱动安装(必须)

驱动下载地址： https://www.nvidia.com/download/index.aspx?lang=en-us

如果没有类似的结果，说明驱动没安装成功。

其中需要关注的是

• Driver Version: 515.65.01 显卡驱动版本

• CUDA Version: 11.7 最高支持CUDA版本

cuda 对驱动的要求 Table 3. CUDA Toolkit and Corresponding Driver Versions 中给出 cuda 对应驱动版本，高版本驱动可向下兼容

2.CUDA 安装(必须)

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

3.cuDNN 安装(可选)

https://developer.nvidia.com/rdp/cudnn-download https://developer.nvidia.com/rdp/cudnn-archive