Step 1: Setting up your GPU#

You will pretty much need a GPU to get started. It is hypothetically possible to proceed with a CPU, but the computations will be really slow.

[1]:

import torch

[2]:

if torch.cuda.is_available():
    print('Hooray, you have a GPU!')
else:
    print("Stop! You should only proceed with a GPU.")

Stop! You should only proceed with a GPU.