PyTorch 2.1.2-foss-2023a-CUDA-12.1.1
There is a newer install of PyTorch
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-2023a-CUDA-12.1.1
To load the module for PyTorch 2.1.2-foss-2023a-CUDA-12.1.1 please use this command on Baskerville:
📋
module load bask-apps/live
module load PyTorch/2.1.2-foss-2023a-CUDA-12.1.1
Bask Apps Env
More Information
For more information visit the PyTorch website.
Dependencies
This version of PyTorch has a direct dependency on: CUDA/12.1.1 cuDNN/8.9.2.26-CUDA-12.1.1 expecttest/0.1.5-GCCcore-12.3.0 FFmpeg/6.0-GCCcore-12.3.0 foss/2023a GMP/6.2.1-GCCcore-12.3.0 magma/2.7.2-foss-2023a-CUDA-12.1.1 MPFR/4.2.0-GCCcore-12.3.0 NCCL/2.18.3-GCCcore-12.3.0-CUDA-12.1.1 networkx/3.1-gfbf-2023a Ninja/1.11.1-GCCcore-12.3.0 numactl/2.0.16-GCCcore-12.3.0 Pillow/10.0.0-GCCcore-12.3.0 protobuf/24.0-GCCcore-12.3.0 protobuf-python/4.24.0-GCCcore-12.3.0 pybind11/2.11.1-GCCcore-12.3.0 Python/3.11.3-GCCcore-12.3.0 Python-bundle-PyPI/2023.06-GCCcore-12.3.0 PyYAML/6.0-GCCcore-12.3.0 SciPy-bundle/2023.07-gfbf-2023a sympy/1.12-gfbf-2023a Z3/4.12.2-GCCcore-12.3.0
Required By
This version of PyTorch is a direct dependent of: PyTorch-Geometric/2.5.0-foss-2023a-PyTorch-2.1.2-CUDA-12.1.1
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.1.2-foss-2022b-CUDA-11.8.0 | live |
| 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 27th September 2024