mars-project / mars

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
https://mars-project.readthedocs.io
Apache License 2.0
2.68k stars 325 forks source link

[BUG] Fix the `GPU DEVICE NOT FOUND` error when deploying Mars on Ray #3333

Closed dlee992 closed 1 year ago

dlee992 commented 1 year ago

What do these changes do?

Mars itself can run with GPU resources, but Mars on Ray or Ray-DAG is infeasible.

This PR will fix the latter issue by handling following parts:

Related issue number

Fixes #3334.