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

时间:2018-06-15 23:13:06

标签: nvidia azure-virtual-machine

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

  

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

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

1 个答案:

答案 0 :(得分:1)

您看到的问题是因为您使用的是D系列VM。只有N系列虚拟机具有GPU。因此,为了使用GPU,您需要选择以下尺寸之一:

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

对于此大小系列,订阅中的vCPU(核心)配额最初在每个区域中设置为0。在可用区域中,您需要request a vCPU quota increase这个家庭。