PyTorch 2.1.2-foss-2022b-CUDA-11.8.0

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

Accessing PyTorch 2.1.2-foss-2022b-CUDA-11.8.0

To load the module for PyTorch 2.1.2-foss-2022b-CUDA-11.8.0 please use this command on Baskerville:

📋 module load bask-apps/live
module load PyTorch/2.1.2-foss-2022b-CUDA-11.8.0

Bask Apps Env



  • cmake 3.26.3
  • lit 16.0.5
  • torch 2.1.2
  • triton 2.1.0

More Information

For more information visit the PyTorch website.


This version of PyTorch has a direct dependency on: CUDA/11.8.0 cuDNN/ expecttest/0.1.4-GCCcore-12.2.0 FFmpeg/5.1.2-GCCcore-12.2.0 foss/2022b GMP/6.2.1-GCCcore-12.2.0 magma/2.6.2-foss-2022b-CUDA-11.8.0 MPFR/4.2.0-GCCcore-12.2.0 NCCL/2.16.2-GCCcore-12.2.0-CUDA-11.8.0 networkx/3.0-gfbf-2022b Ninja/1.11.1-GCCcore-12.2.0 numactl/2.0.16-GCCcore-12.2.0 Pillow/9.4.0-GCCcore-12.2.0 protobuf/23.0-GCCcore-12.2.0 protobuf-python/4.23.0-GCCcore-12.2.0 pybind11/2.10.3-GCCcore-12.2.0 Python/3.10.8-GCCcore-12.2.0 PyYAML/6.0-GCCcore-12.2.0 SciPy-bundle/2023.02-gfbf-2022b sympy/1.11.1-foss-2022b

Other Versions

These versions of PyTorch are available on Baskerville. These will be retained in accordance with our Applications Support and Retention Policy.

Last modified on 26th January 2024