Как настроить виртуальную среду для запуска тестов тензорного потока - PullRequest
Новая
       4

Как настроить виртуальную среду для запуска тестов тензорного потока

0 голосов
/ 04 декабря 2018

Моя цель заключается в настройке виртуальной среды для запуска теста tenorflow , но я получаю сообщение об ошибке при последнем утверждении после выполнения следующего кода.Могу ли я спросить, что я должен сделать, чтобы решить эту проблему?К вашему сведению: я использую Python 3.6.7 только с установленными пакетами pip и virtualenv и CUDA 9.2 / cuDNN7 (все dll в системном пути). 

virtualenv venv
venv\scripts\activate
#now you should be in virtual environment
#install using pip
pip install tf-nightly-gpu

python
import tensorflow

Сообщение об ошибке: 

>>> import tensorflow
Traceback (most recent call last):
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: DLL load failed: The specified module could not be found.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
    from tensorflow.python import pywrap_tensorflow
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 74, in <module>
    raise ImportError(msg)
ImportError: Traceback (most recent call last):
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 58, in <module>
    from tensorflow.python.pywrap_tensorflow_internal import *
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 28, in <module>
    _pywrap_tensorflow_internal = swig_import_helper()
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
    _mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\imp.py", line 243, in load_module
    return load_dynamic(name, filename, file)
  File "C:\MyWork\testarea\testvirtualenv\venv\lib\imp.py", line 343, in load_dynamic
    return _load(spec)
ImportError: DLL load failed: The specified module could not be found.


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.
>

1 Ответ

0 голосов
/ 05 декабря 2018

Нашел ответ сразу после написания вопроса.

Понижение CUDA 9.2 до 9.0 перед копированием cudnn64_7.dll (cuDNN 7.3.1 для CUDA 9.0) в каталог bin CUDA заставило его работать.

В итоге следующий скрипт дал два разных результата издве разные машины. 

virtualenv venv
venv\scripts\activate
pip install tf-nightly-gpu    
python tf_cnn_benchmarks.py --num_gpus=1 --batch_size=32 --model=resnet50 --variable_update=parameter_server

Четырехъядерный процессор i7 7-го поколения с GTX 970

всего изображений в секунду: 78,16

драм1920X с GTX 1080

всего изображений в секунду: 147,77

...