PyTorch 0.4.1

PyTorch is a python package that provides two high-level features:

  • Tensor computation (like numpy) with strong GPU acceleration
  • Deep Neural Networks built on a tape-based autograd system

You can reuse your favourite python packages such as numpy, scipy and Cython to extend PyTorch when needed.

Accessing the software

To load the PyTorch module:

$ module load PyTorch/0.4.1-foss-2018b-Python-3.6.6

Or, to load the module with GPU support:

$ module load PyTorch/0.4.1-fosscuda-2018b-Python-3.6.6

PyTorch is a Python package. It needs to be imported into your Python script to be used:

import torch

<PyTorch commands>

Accessing Previous Versions

Wherever possible, previous versions of this application will be retained for continuity, especially for research projects that require a consistent version of the software throughout the project. Such versions, however, may be unsupported by IT Services or the applications vendor, and may be withdrawn at short or no notice if they can no longer run on the cluster - for example, essential operating system upgrades may be incompatible with old versions.

Previous versions of this application are recorded below:

Known Problems & Limitations


Other Information

The Support Level for this application is An.

Visit PyTorch website for more information regarding this application.

Last modified: 21 December 2018


Professional Services