NVIDIA PhysicsNeMo is an open-source framework for building, training, and fine-tuning Physics-ML models.
What is PhysicsNeMo?
NVIDIA PhysicsNeMo is a toolkit for developing AI enabled physics-ML applications.
With NVIDIA PhysicsNeMo, we aim to provide researchers and industry specialists, with various tools that will help accelerate your development of such models for the scientific discipline of your need.
Visit NVIDIA PhysicsNeMo for more information.
Visit PhysicsNeMo Documentation for documentation.
Running PhysicsNeMo Using Docker
If you have Docker 19.03 or later, a typical command to launch the container with an interactive bash terminal is:
docker run --gpus all --shm-size=1g --ulimit memlock=-1 --ulimit stack=67108864 --runtime nvidia --rm -it nvcr.io/nvidia/physicsnemo/physicsnemo:xx.xx bash
Where xx.xx is the container version. Refer to the latest version here.
Once inside the container, you can clone the PhysicsNeMo repositories from GitHub and use the samples and examples provided to get started with PhysicsNeMo. Refer the Getting Started Guide for more details.
PhysicsNeMo on DGX Cloud (Base Command Platform)
Jobs using the PhysicsNeMo NGC Container on Base Command Platform clusters can be launched either by using the NGC CLI tool or by using the Base Command Platform Web UI. To use the NGC CLI tool, configure the Base Command Platform user, team, organization, and cluster information using the ngc config command as described here.
An example command to launch the container on a single-GPU instance is:
ngc batch run --name "My-1-GPU-PhysicsNeMo-job" --instance dgxa100.80g.1.norm --commandline "sleep 30" --result /results --image "nvcr.io/nvidia/physicsnemo/physicsnemo:xx.xx"
For details on running PhysicsNeMo in Multi-GPU/Multi-Node configuration, refer this Technical Blog and PhysicsNeMo Documentation
For more details on running on DGX Cloud, please refer NVIDIA BCP User Guide
PhysicsNeMo on Public Cloud instances (AWS, GCP, and Azure)
PhysicsNeMo can be used on public cloud instances like AWS, GCP, and Azure. To run PhysicsNeMo,
- Get your GPU instance on AWS, GCP, or Azure.
- Use the NVIDIA GPU-Optimized VMI on the cloud instance. For detailed instructions on setting up VMI refer NGC Certified Public Clouds.
- Once the instance spins up, follow the above instructions to load the PhysicsNeMo Docker container.
For key features, refer NVIDIA PhysicsNeMo Release Notes
License
By pulling and using the container, you accept the terms and conditions of this End User License Agreement.