PyTorch 未检测到 Azure VM 上的 GPU

Question

这是我第一次尝试将 PyTorch 与 GPU 结合使用，我很挣扎。

我正在使用 Azure 虚拟机 (NCasT4_v3-series)，需要手动配置，详细信息如下：https://learn.microsoft.com/en-us/azure/virtual-machines/windows/n-series-driver -设置。这显然是成功的。

然后我必须从这里下载并安装 CUDA 12.1：https://developer.nvidia.com/cuda-downloads。这似乎也有效。

最后我安装了 miniconda 并安装了现有环境（我将现有项目转移到此虚拟机）。然后，我及时运行了在 https://pytorch.org/get-started/locally/ 上找到的代码，即

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

- 也运行成功。

但毕竟，

torch.cuda.is_available()

返回

False

。

我不知道下一步该做什么或任何可能的故障排除步骤。任何帮助将不胜感激。

Answer 1

我已经成功地完成了这项工作，尽管我仍然不知道问题出在哪里。重新登录计算机后，我发现 NVIDIA 设备驱动程序有问题。

我卸载了手动安装的所有内容，而是通过 Azure 门户添加了 NVIDIA 驱动程序，详细信息请参见此处。