Using CUDA with H20 models

David Perez_20831

Morning Miners, is there a way to force the regular H20 models to use the Nvidia GPU and CUDA or is it only through NDJ4??

MartinLiebig

Hi,

ND4J and the new generative ai stuff uses your GPU. The H2O operators will not.

BR,

Martin

David Perez_20831

Thanks Martin, what if  I run RapidMiner over  Docker? 
Will it then use Cuda for all processes?

MartinLiebig

Hi @davepe

there are special docker images which allow CUDA. I am not an expert on the details though.

BR,

Martin