What do I have to do to use GPU acceleration in a prefect fl

Question

What do I have to do to use GPU acceleration in a prefect flow I m using a DaskExecutor with cluster class= dask cuda LocalCUDACluster but I m still getting No cuda image available

Kevin Kho · Answer

Hi < Mikkel Antonsen> what base image are you using for the cluster workers

Mikkel Antonsen · Answer

Uhm not sure I m using a local agent How do I specify cluster workers

Kevin Kho · Answer

Oh my bad just noticed it was LocalCUDACluster Do you have cuda installed then

Mikkel Antonsen · Answer

Yeah

Mikkel Antonsen · Answer

I can run the same function on bare metal with no probs

Mikkel Antonsen · Answer

ooooooh wait a second

Mikkel Antonsen · Answer

The issue was indeed that the prefect environment had cuda 10 and not 11 Thanks for your help

Kevin Kho · Answer

Glad you figured that out