I am trying to run a deep learning job but am having trouble selecting the right tensorflow container. My code works when I utilize the tensorflow GPU notebook but am getting errors when I submit a job using the tensorflow/tensorflow or ufomy/deepo containers. Any help on how to fix this would be appreciated.
@caveman Can you share the errors that you’re seeing? If you know the exact version of the notebook container (the container tag), then you can use that container when submitting the job. Eg the TensorFlow 1.5 Py3 notebook container is pulling down the container from this path: