Как использовать графический процессор Tensorflow с CUDA 10 на RTX 2070?

Через несколько дней я настрою свой новый компьютер на RTX 2070. Я хотел бы использовать графический процессор Tensorflow, но не могу найти совместимые версии графического процессора CUDA и Tensorflow. Насколько я знаю, мне нужна CUDA 10, чтобы воспользоваться дополнительной вычислительной мощностью архитектуры Turing RTX. Но что касается веб-сайта Tensorflow, новейшая версия tf (tensorflow_gpu-1.12.0) работает только с CUDA 9. Я бы предпочел, чтобы все это работало на Windows 10, но если нет другого пути, Linux тоже будет работать.

Где-то в интернете я прочитал о двух слухах: 1. есть какой-то способ скомпилировать неопубликованную версию tf-gpu, которая работает с CUDA 10 2. они опубликуют официальную версию tf-gpu в январе 2019 года (который почти сейчас), который будет поддерживать CUDA 10.

Может ли кто-нибудь подтвердить один из этих слухов (лучше всего указать источник) или сказать мне, как я смогу заставить все это работать?


person Cal Blau    schedule 23.01.2019    source источник


Ответы (1)


Вы правы, что вам нужен cuda 10, и что tensorflow-gpu в настоящее время его не поддерживает. Что вам нужно сделать, так это скомпилировать тензорный поток из исходного кода, как ваш первый слух.

Этапы установки:

  1. Установите CUDA 10 и cuDNN 7.3.1.
  2. Настройте Tensorflow и скомпилируйте его.
  3. Установите пакет .whl с помощью pip

Вот несколько руководств по компиляции tensorflow.

Windows:

https://www.pytorials.com/how-to-install-tensorflow-gpu-with-cuda-10-0-for-python-on-windows/2/

Убунту:

https://medium.com/@saitejadommeti/building-tensorflow-gpu-from-source-for-rtx-2080-96fed102fcca

https://towardsdatascience.com/how-to-make-tensorflow-work-on-rtx-20xx-series-73eb409bd3c0

В качестве альтернативы вы можете найти готовые колеса тензорного потока здесь, пропустив шаг 2:

https://github.com/fo40225/tensorflow-windows-wheel

person Nick    schedule 28.01.2019