TensorFlow Tutorials — AWS Neuron Documentation (original) (raw)

Contents

• Before running a tutorial
• Natural Language Processing
• Utilizing Neuron Capabilities

This document is relevant for: Inf1

TensorFlow Tutorials#

Before running a tutorial#

You will run the tutorials on an inf1.6xlarge instance running Deep Learning AMI (DLAMI) to enable both compilation and deployment (inference) on the same instance. In a production environment we encourage you to try different instance sizes to optimize to your specific deployment needs.

Follow instructions at TensorFlow Tutorial Setup before running a TensorFlow tutorial on Inferentia. We recommend new users start with the ResNet-50 tutorial.

Natural Language Processing#

• Tensorflow 2.x - HuggingFace DistilBERT with Tensorflow2 Neuron [html] [notebook]

Utilizing Neuron Capabilities#

• Tensorflow 2.x - Using NEURON_RT_VISIBLE_CORES with TensorFlow Serving [html]

This document is relevant for: Inf1