CUDA on ARM 1.4

更新

因为疫情的原因,并没有去买 RockPro64 的开发板。在这一段时间中整理了一些相关的内容。所以编号 1.x 放出来了。

这一部分更新的是 NVIDIA GPU 的驱动相关的信息。某天在帮学妹配置环境的时候,上 GeForce 官方网站上去下载,因为实验拿来搞深度学习的卡是 TITAN 系列的(实验室穷啊,谁来赞助赞助我们😂)。然后就看到了 aarch64(arm64) 的显卡驱动,所以这个更新就来写一点相关的内容。

范围

之前 CUDA 10.2 中推出了 aarch64(arm64) 的版本,但是只是预览版本。而且目前的 ARM PC 这个东西,就我知道的,基本都是 ARM-based 的服务器(华为、亚马逊、技嘉这些厂子又产品,还有一家叫 Ampere 的公司,也有,但是主要是处理器)。除了服务器,真正 PC 貌似只有华为一家,鲲鹏 920 处理器的(还贵的很)。 所以我根据 CUDA 10.2 中的一些内容推测 arm64 的驱动应该只会出专业卡或者是计算卡的。但是他们居然出了全系列的!!!

(这一段内容纯属瞎猜)这就意味着使用 OpenGL 或者 Vulkan 源代码开放能在 Linux 上跑的游戏肯能直接从显卡输出。光线追踪也应该是可以用的(你想啊,闭源的驱动为什么要在 ARM 上阉割呢?对吧)。

在 GeForce 官方与 NVIDIA 官方驱动粗略查找了一下,官网上的驱动对于 TITAN 系列,RTX 20 系列,GeForce 16 系列,GeForce 10 系列,GeForce 900 系列, GeForce 700 系列,GeForce 600 系列,Quadro RTX 系列,Quadro 系列 都是支持的(而且应该是没有阉割过的)。但是 V100 和 A100 这样的设备就不得而知了。

方案

目前我已知的最便宜的方案是搞一个基于 RK3399 SOC 的开发板,然后利用这个开发板连接 GPU(参考 CUDA on ARM 1)。