PyTorch 1.10.0-foss-2021a-CUDA-11.3.1

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

Accessing PyTorch 1.10.0-foss-2021a-CUDA-11.3.1

To load the module for PyTorch 1.10.0-foss-2021a-CUDA-11.3.1 please use this command on Baskerville:

module load bask-apps/live
module load PyTorch/1.10.0-foss-2021a-CUDA-11.3.1

Bask Apps Env

live

More Information

For more information visit the PyTorch website.

Dependencies

This version of PyTorch has a direct dependency on: CUDA/11.3.1 cuDNN/8.2.1.32-CUDA-11.3.1 expecttest/0.1.3-GCCcore-10.3.0 FFmpeg/4.3.2-GCCcore-10.3.0 foss/2021a GMP/6.2.1-GCCcore-10.3.0 magma/2.6.1-foss-2021a-CUDA-11.3.1 MPFR/4.1.0-GCCcore-10.3.0 NCCL/2.10.3-GCCcore-10.3.0-CUDA-11.3.1 Ninja/1.10.2-GCCcore-10.3.0 numactl/2.0.14-GCCcore-10.3.0 Pillow/8.2.0-GCCcore-10.3.0 protobuf/3.17.3-GCCcore-10.3.0 protobuf-python/3.17.3-GCCcore-10.3.0 pybind11/2.6.2-GCCcore-10.3.0 Python/3.9.5-GCCcore-10.3.0 PyYAML/5.4.1-GCCcore-10.3.0 SciPy-bundle/2021.05-foss-2021a typing-extensions/3.10.0.0-GCCcore-10.3.0

Required By

This version of PyTorch is a direct dependent of: PyTorch-Geometric/2.0.4-foss-2021a-CUDA-11.3.1

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 11th June 2022