Python Deep Learning Cookbook

How to do it...

1. First, we install PyTorch in our Anaconda environment, as follows:

conda install pytorch torchvision cuda80 -c soumith

If you want to install PyTorch on another platform, you can have a look at the PyTorch website for clear guidance: http://pytorch.org/.

2. Let's import PyTorch into our Python environment:

import torch

3. While Keras provides higher-level abstraction for building neural networks, PyTorch has this feature built in. This means one can build with higher-level building blocks or can even build the forward and backward pass manually. In this introduction, we will use the higher-level abstraction. First, we need to set the size of our random training data:

 batch_size = 32 input_shape = 5 output_shape = 10

4. To make ...