AzureML SDK不使用PyTorch 1.5？

有人使用PyTorch 1.5来使用AzureML SDK（版本1.11和1.12）吗？torch.cuda.is_available（）即使在启用GPU的机器上也返回False。PyTorch 1.3、1.4和1.6的设置完全相同（is_available（）为真）。欢迎指点。这些是（可能）我的Conda环境文件的相关部分，pytorch和azuremlsdk的值根据需要而变化

channels:
  - defaults
  - pytorch
dependencies:
  - python=3.7.3
  - pytorch=1.5.0
  - pip:
      - azureml-sdk==1.12.0

---

感谢

这是PyTorch 1.5和CUDA的一个已知问题，PyTorch在GitHub问题中对此表示认可

他们还没有为这个问题提供正式的解决方案，但他们建议要么更新旧的GPU驱动程序，要么确保安装了支持CPU的PyTorch版本。由于AzureML GPU上的其他PyTorch版本没有遇到此问题，因此GPU驱动程序似乎不是问题所在，因此可能是PyTorch安装的问题

---

尝试将“torchvision==0.6.0”与pytorch=1.5.0一起安装。PyTorch的网站鼓励将1.5.0与torchvision 0.6.0配对：

谢谢-是的，我实际上指定了torchvision==0.6.0，尽管我没有在上面的代码片段中显示它。另一个选项是PyTorch 1.6，在我解决这个问题（也适用于1.6）后，它对我来说运行良好（使用azureml sdk 1.12），按照建议设置环境变量：