Графический процессор Tensorflow для Debian

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


debian tensorflow debian-stretch
person Santhosh    schedule 13.03.2018    source источник

Ответы (2)


arrow_upward
1
arrow_downward

Можно, но хлопотно :)

Я получил Debian 9.3 с Openbox, чтобы он прекрасно работал с Tensorflow 1.6 и Cuda 9.0 + cuDNN 7.0.5.15 (в конечном итоге также с Wavenet).

https://github.com/ella1011/debian_gpu_jungle

person nick88    schedule 14.03.2018

arrow_upward
1
arrow_downward

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

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

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

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

person ash    schedule 14.03.2018