Skip to content
An attempt to train a Tensorflow convnet to provide machine guided image aesthetics
Jupyter Notebook Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

Getting up and running with TensorFlow


sudo apt install nvidia-cuda-toolkit
  • Get cuDNN v4 from -- this will require signing up to the developer program. Use locate cuda.h and locate to find the right place to copy these. The following works on Ubuntu 16.04.
tar xvzf cudnn-7.*
sudo cp cuda/include/cudnn.h /usr/include/cuda
sudo cp cuda/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
sudo chmod a+r /usr/include/cudnn.h /usr/lib/x86_64-linux-gnu/libcudnn*
  • Install using conda
conda create -n tensorflow-gpu python=3.5 anaconda
source activate tensorflow-gpu
pip install jupyter pandas tables matplotlib
pip install --ignore-installed --upgrade ${TF_BINARY_URL}
  • Test that the install has been successful
python -c "import tensorflow"
  • Install bazel: see here
echo "deb [arch=amd64] stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl | sudo apt-key add -
sudo apt update
sudo apt install bazel swig
You can’t perform that action at this time.