Ubuntu14.04にTensorFlowをインストール

by https://www.tensorflow.org/versions/master/get_started/os_setup.html#optional-install-cuda-gpus-on-linux

TensorFolwの必要条件はCuda7.0で、Cuda7.5をインストールしているのでアンインストールしてからCuda7.0を入れる
$sudo apt-get remove cuda
$sudo apt-get remove cuda-toolkit-7-5
$sudo apt-get remove libcudnn4-dev
$CUDA_REPO_PKG=cuda-repo-ubuntu1404_7.0-28_amd64.deb &&
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/$CUDA_REPO_PKG &&
sudo dpkg -i $CUDA_REPO_PKG
$sudo apt-get update
$sudo apt-get install cuda-toolkit-7-0

Cudnnをインストール
https://developer.nvidia.com/rdp/cudnn-archiveからダウンロードしたら
$tar xvzf cudnn-6.5-linux-x64-v2.tgz
$sudo cp cudnn-6.5-linux-x64-v2/cudnn.h /usr/local/cuda/include
$sudo cp cudnn-6.5-linux-x64-v2/libcudnn* /usr/local/cuda/lib64

パスを通す
$sudo gedit ~/.profile
export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64″
export CUDA_HOME=/usr/local/cuda
$source ~/.profile

bazelのインストール
$ sudo add-apt-repository ppa:webupd8team/java
$ sudo apt-get update
$ sudo apt-get install oracle-java8-installer
https://github.com/bazelbuild/bazel/releasesからホームディレクトリにダウンロード
$chmod +x bazel-0.1.1-installer-linux-x86_64.sh
$./bazel-0.1.1-installer-linux-x86_64.sh –user
パスを通す
$sudo gedit ~/.profile
export PATH=”$PATH:$HOME/bin”
$source ~/.profile

その他の依存ライブラリをインストール
$sudo apt-get install python-numpy swig python-dev

configureする
$cd ~/tensorflow
$./configure

pipからTensorFlowをインストール
$sudo pip install –upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.6.0-cp27-none-linux_x86_64.whl

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です