Versions Compared

Old Version 5

changes.mady.by.user Cristian Di Pietrantonio

Saved on

compared with

New Version 6

changes.mady.by.user Cristian Di Pietrantonio

Saved on

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.
Excerpt

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


Column
Panel
titleOn this page:

Table of Contents

...

$ docker pull quay.io/pawsey/pytorch:2.1.2-rocm5.6.0

The container can be also pulled using singularity:

$ singularity pull pytorch.sif docker://quay.io/pawsey/pytorch:2.1.2-rocm5.6.0

 An official AMD container is also available but lacks both support for Cray MPI and some core Python packages, making it unusable on Setonix.

The PyTorch container developed by Pawsey is also available on Setonix as a module installed using SHPC.


Column
width900px
Code Block
languagebash
themeDJango
titleTerminal 1. Checking what version of PyTorch is available on Setonix.
$ module avail pytorch

-------------- /software/setonix/2023.08/containers/views/modules --------------
   pytorch/2.1.0-rocm5.6.0