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 long standing `ValueError` during the GPU/CUDA initialization #3337

Closed dlee992 closed 1 year ago

dlee992 commented 1 year ago

Fixes the long standing error, i.e., ValueError: bytes is not a 16-char string, during gpu/cuda init.

Probably: Fixes: https://github.com/mars-project/mars/issues/3176. Fixes: https://github.com/mars-project/mars/issues/3231. Fixes: https://github.com/mars-project/mars/issues/3325.

Coping from https://github.com/xprobe-inc/xorbits/pull/242. All credits belong to them.

Since Mars hasn't public GPU CI, I would test it on our internal GPU cluster later.

DON'T MERGE it temporarily, since I have to rebase it after https://github.com/mars-project/mars/pull/3336 is merged.