IBM Analog Hardware Acceleration Kit.¶
Instructions on loading the prebuild AIHW toolkit at CCI and running a brief test example. Additional information on using this toolkit can be found at https://analog-ai.mybluemix.net/ and the documentation page https://aihwkit.readthedocs.io/en/latest/analog_ai.html .
These instructions have been tested on our AiMOSx (NPL) cluster.
Prerequisites¶
- Functioning x86 conda environment, conda install directions
- Proxied access the internet enabled, proxy setup directions
Instructions¶
Verify Conda, If you get errors you either don't have conda installed, or have the PPC version loaded instead. See Prerequisite link.
$ conda -V
load AIHW Toolkit conda env¶
$ conda activate /gpfs/u/software/npl-conda/aihwkit_v3_67abd25
Test the simulator¶
$ git clone https://github.com/IBM/aihwkit.git
$ cd aihwkit
$ python examples/08_simple_layer_with_tiki_taka.py
Using the simulator¶
When using the simulator make sure you run on compute nodes not the front end nodes (nplfen01/npl41). Information on using the schedule can be found on our Slurm Page. For short, interactive use see the Interactive section.