Skip to content

pytorch

Back to ROCm containers

Description

PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.

Homepage

https://pytorch.org/

Available Versions on RCAC Clusters

Cluster Versions
BELL 1.8.1-rocm4.2-ubuntu18.04-py3.6, 1.9.0-rocm4.2-ubuntu18.04-py3.6, 1.10.0-rocm5.0-ubuntu18.04-py3.7, 2.4.0-rocm6.3.2-ubuntu24.04-py3.12
NEGISHI 1.8.1-rocm4.2-ubuntu18.04-py3.6, 1.9.0-rocm4.2-ubuntu18.04-py3.6, 1.10.0-rocm5.0-ubuntu18.04-py3.7

Module

You can load the container module by:

module load rocmcontainers
module load pytorch