kayuri / ck-tensorflow

Collective Knowledge repository for evaluating and optimising performance of TensorFlow

ck-tensorflow

Collective Knowledge repository for evaluating and optimising performance of TensorFlow

Installing TensorFlow on Ubuntu

Dependencies

GPU

If you want to use GPU, make sure you have installed CUDA toolkit >=7.0 and cuDNN v2+. If you want to use GPU and pip installation install CUDA toolkit 7.5 and cuDNN v4.

Check if you have CUDA-enabled GPU.

Download and install CUDA.

Download and install cuDNN (Registration required)

More instructions

For pip installation

For python 2.7 install

$ sudo apt-get install  \
    python-dev \
    python-pip

For python 3.5 install

$ sudo apt-get install  \
    python3-dev \
    python3-pip