Skip to content

Instantly share code, notes, and snippets.

@aambert

aambert/tensorflow-Ubuntu.yml

Forked from m3nu/tensorflow-Ubuntu.yml
Created March 4, 2019 11:52
Show Gist options
  • Save aambert/0170288c67df9e6a65ef53b17b1cb699 to your computer and use it in GitHub Desktop.
Save aambert/0170288c67df9e6a65ef53b17b1cb699 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. Manu revised this gist Mar 26, 2018. 1 changed file with 0 additions and 1 deletion.
    1 change: 0 additions & 1 deletion tensorflow-Ubuntu.yml
    View file
    Original file line number Diff line number Diff line change
    @@ -44,7 +44,6 @@
    - swig
    - libcurl3-dev

    http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A
    - name: Install kernel headers
    apt:
    name=linux-headers-{{ ansible_kernel }}
  2. Manu revised this gist Mar 26, 2018. 1 changed file with 1 addition and 8 deletions.
    9 changes: 1 addition & 8 deletions tensorflow-Ubuntu.yml
    View file
    Original file line number Diff line number Diff line change
    @@ -8,7 +8,7 @@
    vars:
    - cuda_network_installer: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
    - cuda_apt_key: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
    - cudnn_runtime: https://files.qmax.us/NN/libcudnn7_7.0.5.15-1%2Bcuda9.0_amd64.deb
    - cudnn_runtime: https://<MY OWN MIRROR HOST>/libcudnn7_7.0.5.15-1%2Bcuda9.0_amd64.deb


    tasks:
    @@ -76,8 +76,6 @@

    - name: Set up work environment
    hosts: all
    vars:
    - project_repo: keras-exercise
    tasks:
    - name: Setup Python environment
    pip: name={{ item }}
    @@ -90,8 +88,3 @@
    - keras
    - h5py
    - pydot

    - name: Clone current project repo
    git:
    repo='https://{{ gitbucket_user }}:{{ gitbucket_pass }}@code.snapdragon.cc/git/manu/{{ project_repo }}.git'
    dest='{{ ansible_env.HOME }}/{{ project_repo }}'
  3. Manu created this gist Mar 26, 2018.
    97 changes: 97 additions & 0 deletions tensorflow-Ubuntu.yml
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,97 @@
    ---
    - name: Install Tensorflow and Keras on ubuntu1604
    hosts: all
    become: yes
    handlers:
    - include: roles/common/handlers/main.yml

    vars:
    - cuda_network_installer: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
    - cuda_apt_key: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
    - cudnn_runtime: https://files.qmax.us/NN/libcudnn7_7.0.5.15-1%2Bcuda9.0_amd64.deb


    tasks:
    - name: Install required packages
    apt: pkg={{ item }} state=installed
    with_items:
    - gcc
    - g++
    - gfortran
    - build-essential
    - linux-image-generic
    - python3-dev
    - python3-pip
    - python3-virtualenv
    - python3-venv
    - libblas-dev
    - liblapack-dev
    - libatlas-base-dev
    - software-properties-common
    - pkg-config
    - apt-utils
    - liblapack-dev
    - libfreetype6
    - libfreetype6-dev
    - libpng12-dev
    - libyaml-dev
    - g++
    - libffi-dev
    - libzmq-dev
    - libzmq1
    - graphviz
    - python3-numpy
    - swig
    - libcurl3-dev

    http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A
    - name: Install kernel headers
    apt:
    name=linux-headers-{{ ansible_kernel }}
    state=present
    update_cache=yes

    - name: Install CUDA network installer
    apt:
    deb={{ cuda_network_installer }}

    - name: Add Nvidia's repo key.
    apt_key:
    url={{ cuda_apt_key }}
    state=present

    - name: Install CUDA packages (1.2G+)
    apt:
    pkg={{ item }}
    update_cache=yes
    state=installed
    with_items:
    - cuda-9-0

    - name: Install cuDNN
    apt:
    deb={{ item }}
    with_items:
    - "{{ cudnn_runtime }}"

    - name: Set up work environment
    hosts: all
    vars:
    - project_repo: keras-exercise
    tasks:
    - name: Setup Python environment
    pip: name={{ item }}
    virtualenv='{{ ansible_env.HOME }}/env'
    virtualenv_command='/usr/bin/python3 -m venv'
    with_items:
    - jupyterlab
    - 'tensorflow==1.5'
    - 'tensorflow-gpu==1.5'
    - keras
    - h5py
    - pydot

    - name: Clone current project repo
    git:
    repo='https://{{ gitbucket_user }}:{{ gitbucket_pass }}@code.snapdragon.cc/git/manu/{{ project_repo }}.git'
    dest='{{ ansible_env.HOME }}/{{ project_repo }}'