Attention: The EBRAINS IAM will be down Monday, the 21st July 2025, from 17.00 CEST (my timezone) for up to 1 hour. This will any affect services requiring an EBRAINS login, we apologise for any inconvenience caused.


Co-Simulation The Virtual Brain Multiscale

Version 27.1 by ldomide on 2022/05/23 22:36

TVB Co-Simulation   

Multiscale: TVB - NEST

Authors: D. Perdikis, L. Domide, J. Mersmann, M. Schirner, P. Ritter 

For more details on TVB see:

Use our Jupyter Hub setup online

We have setup a Jupyter Hub service with tvb-nest as backed already prepared. You will only need an HBP account for accessing this: https://tvb-nest.apps.hbp.eu/

This JupyterHub installation works smoothly with HBP Collab user credentials (login only once at HBP and get access here too). We use a custom Docker Hub tvb-nest image as a backend, and thus a ready to use environment is available immediately, without the need of any local installation or download. This should be the ideal env for demos, presentations or even workshops with tvb-nest.

https://lh6.googleusercontent.com/ytx9eYpMcL3cCScX2_Sxm4CeBW0xbKW3xKsfO2zSId10bW0gw1kiN2_SkexyYBCsF-sKsu0MaJC4cZvGVfQPjMoPBLiePbkvXOZd8BgY3Q0kFzSkRCqQ183lgDQv_6PYoqS3s7uJ

Currently, the users can access 2 folders: TVB-NEST-Examples and Contributed-Notebooks.

The notebooks under TVB-NEST-Examples are public, shared by everyone accessing the instance. Periodically, we will clean all changes under TVB-NEST-Examples folder (by redeploying the pod image), and show the original example notebooks submitted on our Github repo. If users intend to contribute here, they are encouraged to submit changes through Pull Requests (https://github.com/the-virtual-brain/tvb-multiscale)

https://lh6.googleusercontent.com/nnsM0mhXQinmQsJwZwwwe5Sx7f-tZc8t4ELnCh9DwksyVEPUE-jixJTkhoP4l25VKwlDGoXACWtnuxQM9NMOCYbQOzDesgMDlT3sntow___vsEqRVd4OwqMY4BPyBiLJ32BnUbmM

Folder Contributed-Notebooks is not shared. Here, users can experiment with their own private examples. This folder is persisted on restarts in the user HBP Collab personal space. Thus, users will be able to access their work even after a redeploy. (e.g. during a workshop every participant could have in here his own exercise solution).

Running TVB-NEST locally

See more on Github https://github.com/the-virtual-brain/tvb-multiscale  and check this notebook example: https://drive.ebrains.eu/f/b3ea5740fcc34f12af7a/?dl=1

This notebook will be ok to download and try yourself locally, after you have also prepared and launched locally a Docker env: https://hub.docker.com/r/thevirtualbrain/tvb-nest

This is the path recommended for people working closely with tvb-nest. They are able to download it in their local work env and code freely and fast with it.

Running TVB-NEST jobs on CSCS infrastructure from HBP collab

The CSCS and HBP Collab deployment of tvb-nest is a good example to show how tvb-nest can run with an HPC backend. This will be efficient when the simulation jobs are very large. From our experience, with small jobs, the stage-in/out time is considerable, and then the user might be better with just a local run. Also, this deployment requires that the user have an active CSCS personal account. More details on how to use this deployment can be found in this movie: https://drive.google.com/open?id=1osF263FK_NjhZcBJfpSy-F7qkbYs3Q-E

  • Create a collab space of your own
  • Clone and run in your HBP Collab Hub (https://lab.ebrains.eu/) the notebooks from here: https://drive.ebrains.eu/d/245e6c13082f45bcacfa/
    • test_tvb-nest_installation.ipynb  Run the cosimulate_tvb_nest.sh script on the CSCS Daint supercomputer. In this example, basically we are running the installation_test.py file which is in the docker folder.
    • run_custom_cosimulation.ipynb For this example we are using the cosimulate_with_staging.sh script in order to pull the tvb-nest docker image and we are using a custom simulation script (from Github page) which will be uploaded in the staging in phase
    • run_custom_cosimulation_from_notebook.ipynb  This example is running the same simulation as the example above but instead of using an external file with the simulation code we will build a simulation file from a few notebook cells and we will pass this file to the CSCS server.

Few technical details about what we do in these notebooks:

  1. Prepare UNICORE client api.

PYUNICORE client library is available on PYPI. In order to use it you have to install it using:

pip install pyunicore

Next step is to configure client registry and what supercomputer to use

tr = unicore_client.Transport(oauth.get_token())
r = unicore_client.Registry(tr, unicore_client._HBP_REGISTRY_URL)
# use "DAINT-CSCS"  change if another supercomputer is prepared for usage
client = r.site('DAINT-CSCS')

  1. Prepare job submission

In this step we have to prepare a JSON object which will be used in the job submission process.

# What job will execute (command/executable)
my_job['Executable'] = 'job.sh'

# To import files from remote sites to the job’s working directory
my_job['Imports'] = [{
    "From": "https://raw.githubusercontent.com/the-virtual-brain/tvb-multiscale/update-collab-examples/docker/cosimulate_tvb_nest.sh",
    "To" : job.sh
}]

# Specify the resources to request on the remote system
my_job['Resources'] = { 
    "CPUs": "1"}

  1. Actual job submission

In order to submit a job we have to use the JSON built in the previous step and also if we have some local files, we have to give their paths as a list of strings (inputs argument) so the UNICORE library will upload them in the job's working directory in the staging in phase, before launching the job.

job = site_client.new_job(job_description=my_job, inputs=['/path1', '/path2'])
job.properties

  1. Wait until job is completed and check the results

Wait until the job is completed using the following command

# TRUE or FALSE
job.is_running()

Check job's working directory for the output files/directories using

wd = job.working_dir
wd.listdir()

From working job you can preview files content and download files

# Read 'stdout' file
out = wd.stat("stdout")
f = out.raw()
all_lines = f.read().splitlines()
all_lines[-20:]

# Download 'outputs/res/results.npy' file
wd.stat("outputs/res/results.npy").download("results.npy")