| Excerpt |
|---|
PyTorch is an optimised tensor library for deep learning using GPUs and CPUs. |
...
$ docker pull quay.io/pawsey/pytorch:2.1.2-rocm5.6.0
The container can be also pulled using singularity:
...
The PyTorch container developed by Pawsey is also available on Setonix as a module installed using SHPC.
| Column |
|---|
|
| Note |
|---|
Because of software stack deployment policies, container versions deployed on Setonix might be older than what you may find on the online repository. We install new software every six months, roughly. But you are free to pull the latest container in your own space. |
|
To check what version is available, use the the module avail command.
| Column |
|---|
|
| Code Block |
|---|
| language | bash |
|---|
| theme | DJango |
|---|
| title | Terminal 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 |
|
...