如何在GKE中使用较新的CUDA版本?

Question

我目前正在尝试更新我们GKE集群中的一个应用程序。应用程序的更新版本依赖于CUDA 11.2。但是，目前只支持11.0版本(超过1年)。有没有办法强制更新CUDA版本?

此处描述标准流程。

Answer 1

从CUDA 11.0开始，CUDA工具包支持跨小版本的增强版本兼容性。GKE节点映像目前使用Nvidia驱动450.119.04版本，该版本与CUDA 11.x兼容。

因此可以在GKE上部署使用CUDA 11.2构建的应用程序，但这取决于您使用的11.2功能。看到https://docs.nvidia.com/deploy/cuda-compatibility/index.html application-considerations .

使用CUDA 11.0构建的应用程序可以利用所有的11.0特性。