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."
k
Kevin Kho
09/10/2021, 1:51 PM
Hi @Mikkel Antonsen, what base image are you using for the cluster workers?
m
Mikkel Antonsen
09/10/2021, 1:52 PM
Uhm not sure. I'm using a local agent? How do I specify cluster workers?
k
Kevin Kho
09/10/2021, 1:53 PM
Oh my bad, just noticed it was LocalCUDACluster. Do you have cuda installed then?
m
Mikkel Antonsen
09/10/2021, 2:07 PM
Yeah
I can run the same function on bare metal with no probs
ooooooh wait a second
The issue was indeed that the prefect environment had cuda 10 and not 11. Thanks for your help:)