Выполните все шаги, указанные в следующей ссылке.Я получил работу Tensorflow в Ubuntu 18.04
http://www.python36.com/how-to-install-tensorflow-gpu-with-cuda-9-2-for-python-on-ubuntu/
После того, как вы подтвердили установку драйверов nvidia, вам необходимо установить cuDNN, NCCL и Dependencies (libcupti-dev и добавить LD_LIBRARY_PATH).Затем настройте TensorFlow из исходного кода.
wget https://github.com/bazelbuild/bazel/releases/download/0.14.0/bazel-0.14.0-installer-linux-x86_64.sh
chmod +x bazel-0.14.0-installer-linux-x86_64.sh
./bazel-0.14.0-installer-linux-x86_64.sh --user
echo 'export PATH="$PATH:$HOME/bin"' >> ~/.bashrc
и перезапустите переменные среды. Запустите процесс построения TensorFlow, загрузив последний тензор потока.После этого вам нужно построить Tensorflow с помощью Bazel.Этот процесс займет много времени.Это может занять несколько часов.
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
git pull
git checkout r1.8
./configure
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
Чтобы создать whl-файл, выполните следующую команду:
bazel-bin/tensorflow/tools/pip_package/build_pip_package tensorflow_pkg
Чтобы установить тензор потока с помощью pip:
cd tensorflow_pkg
для существующей виртуальной среды:
pip install tensorflow*.whl
После этого вы можете проверить установку TensorFlow, запустив простой пример.