Не могу импортировать gpu версию tenorflow

Question

Я не могу использовать tenorflow, потому что я получаю это сообщение об ошибке, когда пытаюсь импортировать tenorflow:

2020-02-28 09:31:24.742077: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_100.dll'; dlerror: cudart64_100.dll not found
Traceback (most recent call last):
  File "<input>", line 1, in <module>
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2019.3.3\plugins\python-ce\helpers\pydev\_pydev_bundle\pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\tensorflow\__init__.py", line 98, in <module>
    from tensorflow_core import *
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2019.3.3\plugins\python-ce\helpers\pydev\_pydev_bundle\pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\tensorflow_core\__init__.py", line 40, in <module>
    from tensorflow.python.tools import module_util as _module_util
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2019.3.3\plugins\python-ce\helpers\pydev\_pydev_bundle\pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\tensorflow\__init__.py", line 50, in __getattr__
    module = self._load()
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\tensorflow\__init__.py", line 44, in _load
    module = _importlib.import_module(self.__name__)
  File "C:\Users\Maximal\AppData\Local\Programs\Python\Python36\lib\importlib\__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\tensorflow_core\python\__init__.py", line 52, in <module>
    from tensorflow.core.framework.graph_pb2 import *
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2019.3.3\plugins\python-ce\helpers\pydev\_pydev_bundle\pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\tensorflow_core\core\framework\graph_pb2.py", line 7, in <module>
    from google.protobuf import descriptor as _descriptor
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2019.3.3\plugins\python-ce\helpers\pydev\_pydev_bundle\pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)
  File "C:\Users\Maximal\Documents\Python\PyCharm\Projekt1\venv\lib\site-packages\google\protobuf\descriptor.py", line 47, in <module>
    from google.protobuf.pyext import _message
  File "C:\Program Files\JetBrains\PyCharm Community Edition 2019.3.3\plugins\python-ce\helpers\pydev\_pydev_bundle\pydev_import_hook.py", line 21, in do_import
    module = self._system_import(name, *args, **kwargs)

У меня установлен Cuda toolkit 10.0 и соответствующая версия cuDNN, а также Visual Studio. Сначала я получил сообщение:

Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found

Итак, я проверил папку cuda и нашел только cudart64_100.dll. После некоторых исследований я обнаружил, что cudart_101.dll является частью набора инструментов cuda версии 10.1. Поэтому я установил 10.1 с соответствующей версией cuDNN. Но теперь я получаю сообщение:

Could not load dynamic library 'cudart64_100.dll'; dlerror: cudart64_100.dll not found

Это не имеет смысла для меня, и я не нашел способа решить эту проблему. Я использую последнюю версию PyCharm, Python 3.6, Tensorflow-GPU 2.0.0 и Cuda 10.0 и 10.1

Answer 1

На этом этапе я бы просто полностью удалил (например, удалил) оба пакета CUDA (10.0 и 10.1), так как, вероятно, не стоит иметь оба пакета одновременно. После удаления убедитесь, что вы проверили все свои папки, чтобы ничего не осталось.

Затем проверьте, установлены ли последние версии драйверов NVIDIA.

Внимательно следуйте инструкциям по установке Windows. Убедитесь, что вы получили пакет CuDNN, совместимый с установленной вами версией CUDA . Когда вы устанавливаете CuDNN по этой ссылке, это даст вам мастер, где вы сможете выбрать разные версии CuDNN для разных версий CUDA. Выберите один для версии 10.1 и поместите cudnn64_7.dll, cudnn.h и cudnn.lib в нужные папки вашего установочного каталога CUDA соответственно.

Затем я go хотел бы создать новую среду conda, что-то вроде

conda create --name DeepLearning3.6 python=3.6 tensorflow-gpu numpy scipy pandas scikit-learn pandas

(или любые пакеты, которые вам могут понадобиться дополнительно, я обычно использую их как стартовый) отметьте, но убедитесь, что ваша среда , по крайней мере, включает в себя tenorflow-gpu и numpy). Конечно, вы также можете использовать pip и venv, оба должны работать на ваше усмотрение.

Тогда не забудьте указать правильную среду в PyCharm (Ctrl + Alt + S, выберите правильный интерпретатор проекта, который в этом примере будет называться DeepLearning3.6). Попробуйте запустить скрипт еще раз, он должен работать сейчас.