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

Question

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

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

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

Answer 1

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

Шаги установки:

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

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

Окна:

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

Ubuntu:

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