Skip to content

Commit 56e2186

Browse files
committed
Added additional links to Containerised ROCm material.
1 parent 56fb9ab commit 56e2186

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

docs/user-guide/machine-learning.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ A binary install of PyTorch 1.13.1 suitable for ROCm 5.2.3 has been installed ac
2828

2929
This install can be accessed by loading the `pytorch/1.13.1-gpu` module.
3030

31+
!!! note
32+
For GPU, ARCHER2 currently provides access to a legacy version of [ROCm](gpu.md#rocm), `rocm/5.2.3`. This means that users cannot run on GPU a version of PyTorch more recent than 1.13.1. However, it is possible to run PyTorch 2.2.0 via a containerised HPE Cray Programming Environment module, one that features ROCm 5.6.0, see [Containerised ROCm](containers.md/#containerised-rocm) for details.
33+
3134
As DeepCam is an [MLPerf](https://ieeexplore.ieee.org/document/9238612) benchmark, you may wish to base a local python environment on `pytorch/1.13.1-gpu`
3235
so that you have the opportunity to install additional python packages that support MLPerf logging, as well as extra features pertinent to DeepCam (e.g., dynamic learning rates).
3336

docs/user-guide/python.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,8 @@ ensuring that the Python packages will be gathered from the local virtual enviro
137137
The `extend-venv-activate` command becomes available (i.e., its location is placed on the path) only when the ML module is loaded.
138138
The ML modules are themselves based on `cray-python`. For example, `tensorflow/2.12.0` is based on the `cray-python/3.9.13.1` module.
139139

140+
Further info about running ML frameworks on ARCHER2 can be found on the [Machine Learning page](machine-learning.md).
141+
140142
## Conda on ARCHER2
141143

142144
Conda-based Python distributions (e.g. Anaconda, Mamba, Miniconda) are an extremely popular way of installing and

0 commit comments

Comments
 (0)