Tensorflow не будет импортировать в Colab

Я новичок в ML и Colab. Попытка поиграть с проектом по адресу https://github.com/tkarras/progressive_growing_of_gans но с трудом запускает его в Colab.

Когда я запускаю сценарий import_example.py из проекта, я сразу же получаю сообщения об ошибках при загрузке Tensorflow. Поэтому я попытался отступить и просто сделать import tensorflow as tf, Когда я запускаю это, я получаю следующие ошибки. Я установил время выполнения на GPU. Вчера я получил ту же проблему с демо-кодом Colab, но сегодня это работает.

Кто-нибудь может дать мне какие-нибудь указатели? Это похоже на базовые вещи, которые должны просто работать в окружающей среде.

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py in <module>()
     57 
---> 58   from tensorflow.python.pywrap_tensorflow_internal import *
     59   from tensorflow.python.pywrap_tensorflow_internal import __version__

/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py in <module>()
     27             return _mod
---> 28     _pywrap_tensorflow_internal = swig_import_helper()
     29     del swig_import_helper

/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py in swig_import_helper()
     23             try:
---> 24                 _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
     25             finally:

/usr/lib/python3.6/imp.py in load_module(name, file, filename, details)
    242         else:
--> 243             return load_dynamic(name, filename, file)
    244     elif type_ == PKG_DIRECTORY:

/usr/lib/python3.6/imp.py in load_dynamic(name, path, file)
    342             name=name, loader=loader, origin=path)
--> 343         return _load(spec)
    344 

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

During handling of the above exception, another exception occurred:

ImportError                               Traceback (most recent call last)
<ipython-input-8-64156d691fe5> in <module>()
----> 1 import tensorflow as tf

/usr/local/lib/python3.6/dist-packages/tensorflow/__init__.py in <module>()
     22 
     23 # pylint: disable=g-bad-import-order
---> 24 from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
     25 
     26 try:

/usr/local/lib/python3.6/dist-packages/tensorflow/python/__init__.py in <module>()
     47 import numpy as np
     48 
---> 49 from tensorflow.python import pywrap_tensorflow
     50 
     51 from tensorflow.python.tools import component_api_helper

/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py in <module>()
     72 for some common reasons and solutions.  Include the entire stack trace
     73 above this error message when asking for help.""" % traceback.format_exc()
---> 74   raise ImportError(msg)
     75 
     76 # pylint: enable=wildcard-import,g-import-not-at-top,unused-import,line-too-long

ImportError: Traceback (most recent call last):
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "/usr/lib/python3.6/imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "/usr/lib/python3.6/imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory


Failed to load the native TensorFlow runtime.

See https://www.tensorflow.org/install/errors

for some common reasons and solutions.  Include the entire stack trace
above this error message when asking for help.

---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------
Источник

2 ответа

Даже я столкнулся с той же проблемой. Позже я понял, что забыл включить GPU в настройках ноутбука.

Я включил его и установил TensorFlow-GPU (версия для GPU).

Вы можете найти настройки записной книжки в Edit > Notebook Settings.

Вот скриншот

Источник

Проверьте, установлена ​​ли у вас версия tenorflow, которая вам нужна в Colab или нет?! (Возможно, вы установили другую версию tensorflow-gpu и не установили соответствующую версию tensorflow. В этом случае у меня такая же ошибка)

!pip list

Если у вас нет тензорного потока в списке, вы должны установить и тензорный поток, и тензорный поток-gpu следующим образом:

!pip install tensorflow=='enter version here'
!pip install tensorflow-gpu=='enter same version as the previous one here'
Источник

TensorFlow включен по умолчанию в Colab. Итак, я подозреваю, что вы как-то сломали существующую установку.

Вы можете запросить новый бэкэнд, выбрав в меню Runtime -> Reset all runtimes.... Тогда попробуй import tensorflow as tf снова.

Источник

Вам просто нужно внимательно прочитать ошибку:

ПРИМЕЧАНИЕ. Если импорт не выполняется из-за отсутствующего пакета, вы можете вручную установить зависимости, используя! Pip или! Apt.

Попробуйте запустить:

!pip install tensorflow

внутри записной книжки, а затем повторно запустите ячейку с импортом.

Источник
Другие вопросы по тегам