Closed chriseal closed 5 years ago
I think I solved this by running conda create -n h2o4gpuenv -c h2oai -c conda-forge h2o4gpu-cuda10
, which currently isn't mentioned in the README. Maybe someone could update the README?
Also, when I ran the quick check, I got:
import h2o4gpu
import numpy as np
X = np.array([[1.,1.], [1.,4.], [1.,0.]])
model = h2o4gpu.KMeans(n_clusters=2,random_state=1234).fit(X)
model.cluster_centers_
## -- End pasted text --
Out[3]:
array([[1. , 0.5],
[1. , 4. ]])
which is different than what's in the README
My steps were:
And the error message was:
Could I be using the wrong Python version?
Thanks for your help. Running on Ubuntu 18.04.02