...
...
...
...
...
...
...
...
| Excerpt |
|---|
PyTorch is an optimised tensor library for deep learning using GPUs and CPUs. |
| Column | ||||||
|---|---|---|---|---|---|---|
|
...
$ docker pull quay.io/pawsey/pytorch:2.12.20-rocm5.67.03
The container can be also pulled using singularity:
$ singularity pull pytorch.sif docker://quay.io/pawsey/pytorch:2.12.20-rocm5.67.03
An official AMD container is also available but lacks both support for Cray MPI and some core Python packages, making it unusable on Setonix.
...
| Column | |||||||||
|---|---|---|---|---|---|---|---|---|---|
| |||||||||
|