Tensorflow GPU для Debian

Можно ли установить tenorflow GPU в Debian? Я использую Nvidia GTX 1070 Ti и Debian 9.3.0. Я пробовал несколько учебных пособий для Ubuntu, но потерпел неудачу, так как debian не имеет того же репозитория PPA, поддерживаемого Ubuntu, также многие говорили, что добавление репозитория Ubuntu в Debian не рекомендуется

2 ответа

Решение

Это возможно, но это было довольно хлопотно:) Будучи немного расстроенным необходимостью переустанавливать мою систему каждый раз, когда что-то не удавалось установить должным образом или, если это не могло быть должным образом очищено, я сделал несколько сценариев для автоматизации процесса.

Наконец-то я получил Debian 9.3 с Openbox, чтобы он прекрасно работал с Tensorflow 1.6 и Cuda 9.0 + cuDNN 7.0.5.15 (в конце концов также Wavenet). Я использую один графический процессор, но мне не терпится добавить еще один со временем:)

Надеюсь, это поможет вам: https://github.com/ella1011/debian_gpu_jungle

Если что-то неясно, пожалуйста, дайте мне знать:) Удачи!

Вы можете рассмотреть возможность использования docker / nvidia-docker. Бинарные выпуски TensorFlow включают образы докеров, так что вы можете использовать их, чтобы не связываться с вашей локальной средой.

После установки docker / nvidia-docker это будет примерно так:

docker run -it --runtime=nvidia --rm tensorflow/tensorflow:1.6.0-gpu

И, конечно же, вы можете использовать -v флаг, чтобы сделать каталоги на вашем хост-компьютере видимыми для контейнера докера.

Другие вопросы по тегам