Follow the installation instructions below for the deep learning library you are using:

PyTorch installation instructions.