Skip to content

pytorch

Summary

Source: https://pytorch.org

License: LICENSE on github

Path: /software/jupyter/.conda/envs/

Documentation: https://pytorch.org

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 favorite Python packages such as NumPy, SciPy and Cython to extend PyTorch when needed.

Using pytorch

it will generally be sufficient to use the pytorch module

module load maxwell pytorch   # or pytorch/2.3.1 to use a specific version

In rare cases you might need a full setup:

module load maxwell mamba
. mamba-init
mamba activate /software/jupyter/.conda/envs/pytorch-2.3.1