nvidia - nvidia-smi 命令可以与 nvidia 驱动程序 microsoft azure dsvm 通信

Question

在创建并启动数据科学虚拟机并通过 ssh 连接之后，我尝试使用nvidia-smi来查看内置的 nvidia 和 cuda 是否正常工作。返回的消息已读

NVIDIA-SMI 失败，因为它无法与 NVIDIA 驱动程序通信。确保已安装并运行最新的 NVIDIA 驱动程序。

这些应该是 vm 的一部分，但是当我尝试运行我创建的程序时，使用的是本地计算机的默认 CPU，而不是 vm 的 GPU。我项目的最终目标是运行一个对象检测模型，其性能从我糟糕的 11 秒/图像中加速，所以我想我会使用一个 vm 并利用它的计算能力。然而，这似乎不是最好的选择，所以如果其他人有一些建议，我将不胜感激。

标签： nvidiaazure-virtual-machine

Answer 1

您看到的问题是因为您使用的是 D 系列 VM。只有 N 系列 VM 有 GPU。因此，为了利用 GPU，您需要选择以下尺寸之一：

https://docs.microsoft.com/en-us/azure/virtual-machines/windows/sizes-gpu

对于此大小系列，您订阅中的 vCPU（核心）配额最初在每个区域中设置为 0。您将需要在可用区域中为该系列请求增加 vCPU 配额。