Skip to content

Commit

Permalink
Add docs for ABCI3.0.
Browse files Browse the repository at this point in the history
  • Loading branch information
KanaiYuma-aist committed Feb 3, 2025
1 parent 5450af7 commit 50b6013
Showing 1 changed file with 41 additions and 0 deletions.
41 changes: 41 additions & 0 deletions docs/source/user_guide/torch.rst
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,47 @@ Writing a simple training script
Running inference
-----------------

To run aiaccel on ABCI3.0, you need an environment with Python 3.10.
This guide explains how to set up the environment using Singularity.

.. note::

For details on how to use Singularity, please refer to the following documentation:
https://docs.abci.ai/v3/en/containers/

Create the following Singularity definition file:

.. code-block:: bash
:caption: aiaccel_env.def
BootStrap: docker
From: python:3.10
%post
pip install --upgrade pip
# aiaccel env
pip install aiaccel[torch]@git+https://github.com/aistairc/aiaccel.git@develop/v2
# torch/MNIST example env
pip install torchvision
Use the Singularity definition file to build a Singularity image file:

.. code-block:: bash
singularity build aiaccel.sif aiaccel_env.def
Use the Singularity image file to execute aiaccel:

.. code-block:: bash
singularity exec --nv aiaccel.sif python -m aiaccel.torch.apps.train $wd/config.yaml --working_directory $wd
Writing a DDP training script
-----------------------------

Expand Down

0 comments on commit 50b6013

Please sign in to comment.