awslabs / datawig

Imputation of missing values in tables.
Apache License 2.0
475 stars 69 forks source link

Run on GPU #158

Open bbb801 opened 2 years ago

bbb801 commented 2 years ago

Dear Sir, May I know how to run the tool on GPU? I have installed the GPU version of Datawig. I always get an error. Thank you.

RuntimeError: simple_bind error. Arguments: __empirical_risk_instance_weight__: (16, 1) numerical_features-zHE8kjaHDe: (16, 2) Current employment status: (16, 1) Traceback (most recent call last): File "../include/mxnet/base.h", line 458 CUDA: Check failed: e == cudaSuccess (30 vs. 0) : unknown error

image image