What do I have to do to use GPU acceleration in a ...
# prefect-serverm
Mikkel Antonsen
09/10/2021, 12:28 PMWhat 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 PMUhm 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 PMYeah
Mikkel Antonsen
09/10/2021, 2:07 PMI can run the same function on bare metal with no probs
Mikkel Antonsen
09/10/2021, 2:07 PMooooooh wait a second
Mikkel Antonsen
09/13/2021, 8:19 AMThe issue was indeed that the prefect environment had cuda 10 and not 11. Thanks for your help:)
k
Kevin Kho
09/13/2021, 3:49 PM
Glad you figured that out!
2 Views