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
Extensions
- 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.
Dependencies
This version of PyTorch has a direct dependency on: CUDA/11.8.0 cuDNN/8.7.0.84-CUDA-11.8.0 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.
Version | Bask Apps Env |
---|---|
2.0.1-foss-2022a-CUDA-11.7.0 | live |
1.10.0-foss-2021a-CUDA-11.3.1 | live |
1.8.1-fosscuda-2020b | live |
1.8.1-fosscuda-2020b | test |
Last modified on 26th January 2024