This is how I managed to build TensorFlow 2.0 on Ubuntu 18.04 (x86_64) with Bazelisk:
$ sudo apt update
$ sudo apt full-upgrade
$ sudo apt install curl
# Install Bazelisk.
$ sudo curl -Lo /usr/local/bin/bazel https://github.com/bazelbuild/bazelisk/releases/download/v1.1.0/bazelisk-linux-amd64
$ sudo chmod +x /usr/local/bin/bazel
| # WARNING: These steps seem to not work anymore! | |
| #!/bin/bash | |
| # Purge existign CUDA first | |
| sudo apt --purge remove "cublas*" "cuda*" | |
| sudo apt --purge remove "nvidia*" | |
| # Install CUDA Toolkit 10 | |
| wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb |
| #!/bin/bash | |
| ### steps #### | |
| # Verify the system has a cuda-capable gpu | |
| # Download and install the nvidia cuda toolkit and cudnn | |
| # Setup environmental variables | |
| # Verify the installation | |
| ### | |
| ### to verify your gpu is cuda enable check |
| """An example of how to use tf.Dataset in Keras Model""" | |
| import tensorflow as tf # only work from tensorflow==1.9.0-rc1 and after | |
| _EPOCHS = 5 | |
| _NUM_CLASSES = 10 | |
| _BATCH_SIZE = 128 | |
| def training_pipeline(): | |
| # ############# | |
| # Load Dataset |
| '''Train MNIST with tfrecords yielded from a TF Dataset | |
| In order to run this example you should first run 'mnist_to_tfrecord.py' | |
| which will download MNIST data and serialize it into 3 tfrecords files | |
| (train.tfrecords, validation.tfrecords, and test.tfrecords). | |
| This example demonstrates the use of TF Datasets wrapped by a generator | |
| function. The example currently only works with a fork of keras that accepts | |
| `workers=0` as an argument to fit_generator, etc. Passing `workers=0` results | |
| in the generator function being run on the main thread (without this various |
| /* Compile with: g++ -Wall –Werror -o shell shell.c */ | |
| #include <stdio.h> | |
| #include <stdlib.h> | |
| #include <string.h> | |
| #include <ctype.h> | |
| #include <unistd.h> | |
| #include <sys/types.h> | |
| #include <sys/wait.h> |
