PyTorch无法检测Azure VM上的GPU

35g0bw71  于 2023-11-19  发布在  其他
关注(0)|答案(1)|浏览(171)

这是我第一次尝试在GPU上使用PyTorch,我很挣扎。
我使用的是Azure虚拟机(NCasT4_v3系列),需要手动配置,详细说明如下:https://learn.microsoft.com/en-us/azure/virtual-machines/windows/n-series-driver-setup。这显然是成功的。
然后我不得不从这里下载并安装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
我不知道下一步该做什么,也不知道任何可能的故障排除步骤。任何帮助都将不胜感激。

mctunoxg

mctunoxg1#

我已经设法让这个工作,虽然我仍然没有任何洞察到什么问题是。在登录回到机器上,我发现一个问题的NVIDIA设备驱动程序。
我重新安装了手动安装的所有内容,并通过Azure Portal添加了NVIDIA驱动程序,详情如下。

相关问题