GPU.md

📚 GPU Mode

In order to enforce the GPU execution mode, set this global variable:

GrAI.Opti.GPU = true

Optionally, you can also set the following global variable in order to precise the list of external GPU devices to use (first in the list will be used with top priority):

GrAI.Opti.gpuNamedPriority = ["<GPU name1>", "<GPU name2>", ...]

Then, in order to initialize the GPU kernel, run:

_ = MetalKernel.get

Next Chapter

Previous chapter: Layer.
Next chapter: Model.