Ошибки с tenorflow-gpu 2.0alpha0

Question

Я установил тензор-gpu 2.0alpha0, cudatoolkit 10.0.130 и cudnn 7.5 в Windows 10.

TensorFlow 2.0 хорошо работает с процессором, но сталкивается с ошибками при работе с графическим процессором.

Сообщения об ошибках приведены ниже:

2019-04-02 23:47:38.646661: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
2019-04-02 23:47:38.666653: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Could not dlopen library 'nvcuda.dll'; dlerror: nvcuda.dll not found
2019-04-02 23:47:38.666842: E tensorflow/stream_executor/cuda/cuda_driver.cc:318] failed call to cuInit: UNKNOWN ERROR (303)

Я думаю, что главная проблема - «Не удалось открыть библиотеку nvcuda.dll».

Однако я установил последнюю версию драйвера NVIDIA (версия 419.67), и файл nvcuda.dll находится в C: \ Windows \ System32 \ nvcuda.dll.

Answer 1

Это метод, который не работает в Tensorflow:

    port::StatusOr<void*> GetDsoHandle(const string& name, const string& version) {
        auto filename = port::Env::Default()->FormatLibraryFileName(name, version);
        void* dso_handle;
        port::Status status =
          port::Env::Default()->LoadLibrary(filename.c_str(), &dso_handle);
        if (status.ok()) {
            LOG(INFO) << "Successfully opened dynamic library " << filename;
            return dso_handle;
    }

По сути, это означает, что Tensorflow не может найти nvcuda из переменной PATH в вашей среде Windows. Чтобы добавить переменную среды, сделайте следующее:

Windows 10 и Windows 8

1. В Поиске найдите и выберите: Система (Панель управления)
2. Нажмите на ссылку Дополнительные параметры системы.
3. Нажмите Переменные среды. В разделе Системные переменные найдите переменную среды PATH и выберите ее. Нажмите Редактировать.
4. Если переменная среды PATH не существует, нажмите кнопку Создать.
5. В окне Изменить системную переменную (или новую системную переменную) добавьте значение пути C: \ Windows \ System32 \ nvcuda.dll. Нажмите ОК. Закройте все оставшиеся окна, нажав ОК.

PS: Убедитесь, что вы установили cuDNN