PyTorch¶
Tensors and Dynamic neural networks in Python with strong GPU acceleration.PyTorch is a deep learning framework that puts Python first.
https://pytorch.org/
Available modules¶
The overview below shows which PyTorch installations are available per target architecture in EESSI, ordered based on software version (new to old).
To start using PyTorch, load one of these modules using a module load command like:
(This data was automatically generated on Thu, 12 Sep 2024 at 12:39:49 UTC)
| aarch64/generic | aarch64/neoverse_n1 | aarch64/neoverse_v1 | x86_64/generic | x86_64/amd/zen2 | x86_64/amd/zen3 | x86_64/amd/zen4 | x86_64/intel/haswell | x86_64/intel/skylake_avx512 | |
|---|---|---|---|---|---|---|---|---|---|
| PyTorch/2.1.2-foss-2023a | x | x | x | x | x | x | x | x | x |