Как правильно запускать TensorFlow 2.0 на GPU?

#tensorflow #anaconda #tensorflow2.0

#tensorflow #anaconda #tensorflow2.0

Вопрос:

Я использовал эти команды, чтобы сделать это на моем ПК с Windows 10 без AVX, поддерживаю только SSE4.2, GTX1060 6GB с последними версиями драйверов:

 conda create --name tensorflow-2.0 python=3.6
activate tensorflow-2.0
pip install tf-nightly-gpu-2.0-preview
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
  

Все было хорошо, пока я не попытался использовать TensorFlow:

 import tensorflow as tf

ImportError: Traceback (most recent call last):
  File "C:UsersFrutAnaconda3libsite-packagestensorflowpythonpywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:UsersFrutAnaconda3libsite-packagestensorflowpythonpywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:UsersFrutAnaconda3libsite-packagestensorflowpythonpywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "C:UsersFrutAnaconda3libimp.py", line 242, in load_module
    return load_dynamic(name, filename, file)
  File "C:UsersFrutAnaconda3libimp.py", line 342, in load_dynamic
    return _load(spec)
ImportError: DLL load failed: Не найден указанный модуль.


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.
  

Я попробовал другой способ:

 conda remove -n tensorflow-2.0 --all
  

Удаленная папка «tensorflow-2.0» в «Anaconda3 envs»

 conda create --name tensorflow-2.0 python=3.6
conda activate tensorflow-2.0 amp;amp; conda install -c anaconda pip
  

После этого ошибок не было, но активная среда по-прежнему «базовая *»

 activate tensorflow-2.0
  

Теперь активным env является tensorflow-2.0

 conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
pip install tensorflow-gpu==2.0.0-alpha0
  

И результат — та же ошибка:

 import tensorflow as tf
Failed to load the native TensorFlow runtime
  

Ответ №1:

В документации в настоящее время рекомендуется использовать эту команду для установки Tensorflow2.0 с поддержкой GPU:

 $ pip install tensorflow-gpu==2.0.0-alpha0
  

Чтобы установить его внутри conda (при условии, что у вас оборудование с поддержкой CUDA), вы должны сначала установить pip внутри среды, чтобы:

 $ conda activate your_environment amp;amp; conda install -c anaconda pip
  

После этого выполните команду pip. На вашем месте я бы уничтожил эту среду на всякий случай и создал ее с нуля.

Комментарии:

1. Спасибо. Я сделал то, что вы сказали, но по-прежнему безуспешно. Я обновил сообщение.

2. Что ж, боюсь, тогда вам может не повезти. Некоторые другие вещи, приходящие на ум: обновление cuda до 10.0 (или понижение до 9.0, если у вас CUDA 10.0) и обеспечение того, чтобы CUDA_PATH переменная среды была установлена правильно. Правильно ли работает версия CPU?

3. Что CUDA_PATH следует использовать при установке CUDA с помощью команд conda? Я не использую версию TF для процессора. Это гораздо лучше делать с помощью colab GPU или даже TPU от Google.

4. Проверьте здесь и убедитесь, что ваша система поддерживает CUDA. Это инструкции для Windows, я недостаточно знаком с этой ОС, поэтому не могу с этим помочь. И вам следует проверить версию Tensorflow для процессора, чтобы можно было точно определить, связана ли эта проблема с GPU или с Tensorflow.