PyTorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration. PyTorch is a deep learning framework that puts Python first.

homepage: https://pytorch.org/

Available installations

PyTorch version	Supported CPU targets	Supported GPU targets	EESSI version	Module
2.1.2	generic: aarch64, x86_64 Arm: a64fx, neoverse_n1, neoverse_v1, nvidia/grace AMD: zen2, zen3, zen4 Intel: haswell, skylake_avx512, sapphirerapids, icelake, cascadelake	(none)	2023.06	PyTorch/2.1.2-foss-2023a