TensorFlow GPU не работает, как его установить? - PullRequest
Новая
       5

TensorFlow GPU не работает, как его установить?

0 голосов
/ 01 марта 2019

Я недавно начал изучать тензор потока и решил переключиться на версию GPU, потому что она намного быстрее, но я не могу ее импортировать, она всегда выдает ту же ошибку.

Я уже пробовал:

  • Установка его с помощью pip, в python 3.6.8, cuda 10 и самой последней cuDNN для cuda 10
  • Я попытался переустановить python, CUDA и cuDNN
  • Попробовал установить Visual Studio и установить CUDA 9 и cuDnn
  • Я попытался установить последнюю Anaconda, создал env «по умолчанию» и еще один в python 3.6 (также пытался в 3.5), pip install tenorflow-gpu в обоихслучаи
  • моей последней попыткой было следовать учебному пособию на YouTube, я сделал именно так, как продемонстрировано (https://www.youtube.com/watch?v=KZFn0dvPZUQ)

Все, что я пробовал, возвращало ту же ошибку.

Traceback: https://pastebin.com/KMEsZAmq

Полный код: https://pastebin.com/7tS0Rd5S (работал на версии процессора) 

    .

Мои характеристики:

  • i5-8400
  • 8 ГБ RAM
  • GTX 1060 6 ГБ
  • W10 home x64

1 Ответ

0 голосов
/ 01 марта 2019

просто посмотрите здесь: https://www.tensorflow.org/install/gpu

Tensorflow поддерживает CUDA 9.0, вам нужно понизить CUDA или использовать один из образов docker tenorflow: https://www.tensorflow.org/install/docker

через докерон не будет использовать ваши драйверы CUDA

...