Any solution for this: https://github.com/PrefectHQ/prefect/issues/4737 ?

Try downloading and reading this. This is the most comprehensive answer for now

Thanks it worked. Will it be the same in Prefect 2.0 ?

Maybe? Because 2.0 based off docker-py also. We would need them to add the gpu support in their library