ai

PyTorch

https://pytorch.org/

Package description

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

Use latest version

module load PyTorch

Use specific version

module load PyTorch/1.12.0-foss-2022a-CUDA-11.7.0

module load PyTorch/1.12.0-foss-2022a

---

PyTorch-bundle

https://pytorch.org/

Package description

PyTorch with compatible versions of official Torch extensions.

Use latest version

module load PyTorch-bundle

Use specific version

module load PyTorch-bundle/1.12.1-foss-2022a-CUDA-11.7.0

---

colossalai

https://colossalai.org/

Package description

Colossal-AI: A Unified Deep Learning System for Big Model Era

Use latest version

module load colossalai

Use specific version

module load colossalai/0.1.8-foss-2021a-CUDA-11.3.1

---