Skip to content
NAISS Logo Allocations Support Training Software naiss.se

Pytorch

General information

Installed versions

Resource Version
Arrhenius 2.9.1

For software provided as an Lmod module, use the documentation search function and search for Lmod module to learn how to load it in your environment.

PyTorch is an open-source deep learning library for Python. PyTorch provides two high-level features:

  • General tensor computations with strong GPU acceleration
  • Dynamically built computation graphs with automatic differentiation for backpropagation