Как использовать Tensorflow GPU с CUDA 10 на RTX 2070? - PullRequest
Новая
       118

Как использовать Tensorflow GPU с CUDA 10 на RTX 2070?

0 голосов
/ 23 января 2019

через несколько дней я настрою свой новый компьютер с 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.

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

1 Ответ

0 голосов
/ 28 января 2019

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

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

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

Вот несколько учебных пособий для компиляции тензорного потока.

Windows:

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

...