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?

To train with a GPU, you must install CUDA toolkit 11.8 and cuDNN 8.7. 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.

    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.

  2. Install CUDA toolkit 11.8 and cuDNN 8.7.
  3. Update the "path" environment variable to include the following directories (this may be different on your machine).
    For example, on a Windows PC, the following paths need to be added.
    • C:\Program Files\NVIDIA\CUDNN\v8.7\bin
    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
    • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp
  4. Relaunch HyperMesh.

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.