Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
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
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.