Sagemaker Studio notebook - no module named 'tensorflow' when chosen image type "Tensorflow 2.6 Python 3.8 GPU optimized"

0

Hi,

I have created a simple notebook on SageMaker Studio using the Image "Tensorflow 2.6 Python 3.8 GPU optimized". But when I try to run simple statement viz. "import tensorflow", I am getting the error "no module named 'tensorflow'".

I tried to install 'tensorflow' package using pip from the terminal attached to the image. But it shows the message "requirement already satisfied".

Am I missing anything here? Please help.

Thanks in advance, Praveen

gefragt vor 3 Jahren973 Aufrufe
1 Antwort
1
Akzeptierte Antwort

From the question, I understand that you are trying to use a TensorFlow 2.6 Python 3.8 kernel in SageMaker Studio, but you are unable to import tensorflow.

The service team are aware of this issue and are actively working on a fix.

Mitigation Option

A) If your use case is version flexible, version other than 2.6 should work.

B) If not, you can try the following as workaround

  1. Open a notebook using a Tensorflow 2.6 Python 3.8 kernel
  2. Execute the following line in a notebook cell: !sed -i 's|^ *"python",| "/usr/local/bin/python",|g' /usr/local/share/jupyter/kernels/python3/kernel.json
  3. Stop the kernel
  4. Re-attach the kernel to your notebook.

Hope it helps!

SUPPORT-TECHNIKER
beantwortet vor 3 Jahren
profile picture
EXPERTE
überprüft vor 6 Monaten
  • Thanks for informing that it's a known issue. Are there any other known issues of SageMaker Studio documented in a webpage? Please send me the link if it exists.

    The solution mentioned in option #B didn't work for me. Tried for couple of times, but still I am getting same error.

    I went for option #A and it's working fine as for Tensorflow.

    Thanks for the help!!

  • Option #B also working fine for me. Thanks for the workaround!!

Du bist nicht angemeldet. Anmelden um eine Antwort zu veröffentlichen.

Eine gute Antwort beantwortet die Frage klar, gibt konstruktives Feedback und fördert die berufliche Weiterentwicklung des Fragenstellers.

Richtlinien für die Beantwortung von Fragen