PyTorch 2.0.1-foss-2022a-CUDA-11.7.0

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.0.1-foss-2022a-CUDA-11.7.0

To load the module for PyTorch 2.0.1-foss-2022a-CUDA-11.7.0 please use this command on Baskerville:

📋 module load bask-apps/live
module load PyTorch/2.0.1-foss-2022a-CUDA-11.7.0

Bask Apps Env

live

Extensions

• cmake 3.26.3
• lit 16.0.5
• torch 2.0.1
• triton 2.0.0

More Information

For more information visit the PyTorch website.

Dependencies

This version of PyTorch has a direct dependency on: CUDA/11.7.0 cuDNN/8.4.1.50-CUDA-11.7.0 expecttest/0.1.3-GCCcore-11.3.0 FFmpeg/4.4.2-GCCcore-11.3.0 foss/2022a GMP/6.2.1-GCCcore-11.3.0 magma/2.6.2-foss-2022a-CUDA-11.7.0 MPFR/4.1.0-GCCcore-11.3.0 NCCL/2.12.12-GCCcore-11.3.0-CUDA-11.7.0 networkx/2.8.4-foss-2022a Ninja/1.10.2-GCCcore-11.3.0 numactl/2.0.14-GCCcore-11.3.0 Pillow/9.1.1-GCCcore-11.3.0 protobuf/3.19.4-GCCcore-11.3.0 protobuf-python/3.19.4-GCCcore-11.3.0 pybind11/2.9.2-GCCcore-11.3.0 Python/3.10.4-GCCcore-11.3.0 PyYAML/6.0-GCCcore-11.3.0 SciPy-bundle/2022.05-foss-2022a sympy/1.11.1-foss-2022a

Required By

This version of PyTorch is a direct dependent of: Horovod/0.28.0-foss-2022a-CUDA-11.7.0-PyTorch-2.0.1 MONAI/1.2.0-foss-2022a-CUDA-11.7.0 PyTorch-Lightning/2.0.2-foss-2022a-CUDA-11.7.0 torchvision/0.15.2-foss-2022a-CUDA-11.7.0

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 2nd June 2023