How to run CuDNNLSTM on JupyterLab within Azure Machine Learning?

Bhekisizwe Mthethwa 1 Reputation point
2020-07-22T07:55:43.68+00:00

Good day Everyone

I am trying to train an LSTM based recurrent neural network using Azure Machine Learning JupyterLab. I have setup the VM instance to use a 6 core GPU. However, when i try to train my recurrent network using the efficient GPU based CuDNNLSTM network i get a "ModuleNotFoundError: no module named tensorflow.contrib". How can i rectify this so that i can be able to run CuDNNLSTM based code on my GPU?

Azure Machine Learning
Azure Machine Learning
An Azure machine learning service for building and deploying models.
2,848 questions
{count} votes

1 answer

Sort by: Most helpful
  1. Bhekisizwe Mthethwa 1 Reputation point
    2020-07-23T07:03:36.42+00:00

    Thank you. I am using the Microsoft Azure Machine Learning JupyterLab notebook so it comes pre-installed with all the packages and dependencies. So i have no idea what they are using unless if you know where i can check the versions used by Microsoft Azure?

    0 comments No comments

Your answer

Answers can be marked as Accepted Answers by the question author, which helps users to know the answer solved the author's problem.