显卡驱动与CUDA版本对照
| Driver Version | CUDA Version | docker image |
|---|---|---|
| 510 | 11.6 | docker pull nvidia/cuda:11.6.2-runtime-ubuntu20.04 |
| 525 | 12.0 | docker pull nvidia/cuda:12.0.0-runtime-ubuntu20.04 |
| 530 | 12.1 | docker pull nvidia/cuda:12.1.0-runtime-ubuntu20.04 |
| 535 | 12.2 | docker pull nvidia/cuda:12.2.0-runtime-ubuntu20.04 |
所在服务器安装对应的显卡驱动（一般都有了），需要使用对应的 cuda image与其对应
对外prover docker统一使用ubuntu 20.04
- base:从 CUDA 9.0 开始，包含部署预构建 CUDA 应用程序的最低限度（libcudart）。如果您想手动选择要安装的 CUDA 软件包，请使用此映像。
- runtime:通过添加 CUDA 工具包中的所有共享库来扩展基础映像。如果您有使用多个 CUDA 库的预构建应用程序，请使用此映像。
- devel:通过添加编译器工具链、调试工具、标头和静态库来扩展运行时映像。使用此映像从源代码编译 CUDA 应用程序