Skip to content

deepspeed

Back to ROCm containers

Description

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

Homepage

https://www.deepspeed.ai

Available Versions on RCAC Clusters

Cluster Versions
BELL rocm4.2_ubuntu18.04_py3.6_pytorch_1.8.1, rocm6.1_ubuntu20.04_py3.9_pytorch_2.1.2
NEGISHI rocm4.2_ubuntu18.04_py3.6_pytorch_1.8.1

Module

You can load the container module by:

module load rocmcontainers
module load deepspeed