How to use CuDNNLSTM in Azure JupyterLab? RRS feed

  • Question

  • 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?

    Tuesday, July 21, 2020 10:52 PM