GPU support

In order to use your GPU to run NDK simulations you will have to install the NVIDIA HPC SDK.

Warning

Make sure the HPC SDK environment variables are exported.

You will only have to export one environment variable to enable the GPU support for NDK:

export PLATFORM="nvidia-acc"

Now when running NDK simulations you should be able to see platform=nvidiaX in the execution output:

import neurotechdevkit as ndk

scenario = ndk.scenarios.built_in.Scenario2_2D()
scenario.make_grid()
scenario.compile_problem()
result = scenario.simulate_steady_state()
result.render_steady_state_amplitudes()

Output:

...

Operator `acoustic_iso_state` instance configuration:
     * subs={h_x: 0.0005, h_y: 0.0005}
     * opt=advanced
     * compiler=pgcc
     * language=openacc
     * platform=nvidiaX

...

Warning

Simulations with high memory requirement (e.g. 3D simulations) may not fit in the GPU and running them with GPU acceleration might crash the simulation.