Computing and Resources

Is a GPU required?

A CPU can be used to train a model, but it will be slower than a GPU.

What GPUs are supported?

This requires a NVIDIA GPU that has a compute capability > 6.0. See the NVIDIA website for more information on the compute capability of specific hardware.

How do I enable NVIDIA GPUs for PhysicsAI training?

  1. Verify the GPU is supported.
  2. Launch and use HyperMesh.

    Supported GPUs will be automatically used.

What effect does GPU and dataset size have on training times?

The quality of computing resources will affect training time. Better hardware can improve run times, but improvements from CPU to GPU can be substantial. Regardless, training time is linearly proportional to dataset size. See the table below for a representative example:
Hardware 10 Result Files 50 Result Files 100 Result Files
Laptop CPU 44 m 3 h 35 m 7 h 54 m
HPC CPU 34 m 2 h 46 m 5 h 10 m
GPU 3 m 16m 33 m

Can I train on an HPC?

Yes, more information is available here.