CODEWITHSUNDEEP
pytorchazure-machine-learning-service
David Carter
David Carter

Reputation: 21

AzureML SDK not working with PyTorch 1.5?

Has anyone got PyTorch 1.5 to work with the AzureML SDK (versions 1.11 and 1.12)? torch.cuda.is_available() returns False even on GPU-enabled machines. Exactly the same setup works fine (is_available() is True) with PyTorch 1.3, 1.4 and 1.6. Any pointers welcome. These are the (possibly) relevant parts of my Conda environment file, with the values of pytorch and azureml-sdk varied as required.

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

Thanks

Upvotes: 2

Views: 236

Answers (1)

dpeck
dpeck

Reputation: 11

This is a known issue with PyTorch 1.5 and CUDA and is acknowledged by PyTorch in this GitHub issue.

They haven't provided an official solution to the issue, but they recommend either updating old GPU-drivers or making sure you have a CPU-enabled version of PyTorch installed. Since you're not experiencing this problem with other PyTorch versions on AzureML GPUs, GPU drivers don't seem to be the issue, so it's probably the PyTorch installation.

Try installing "torchvision==0.6.0" along with your pytorch=1.5.0. PyTorch's site encourages pairing 1.5.0 with torchvision 0.6.0: https://pytorch.org/get-started/previous-versions/

Upvotes: 1

Related Questions