I'm getting this error on a fresh EC2 machine (p2.xlarge, ubuntu 16.04) on which I've installed cuda, python3, numpy and libKMCUDA with pip.
(venv) ubuntu@ip-172-31-17-253:~/code/km-test$ python test.py
reassignments threshold: 100
compute capability mismatch for device 0: wanted 6.1, have 3.7
>>>> you may want to build kmcuda with -DCUDA_ARCH=37 (refer to "Building" in README.md)
Traceback (most recent call last):
File "test.py", line 10, in <module>
centroids, assignments = kmeans_cuda(arr, 4, verbosity=1, seed=3)
ValueError: No such CUDA device exists
Here's how I installed CUDA:
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
apt-get update
apt-get install cuda
I'm getting this error on a fresh EC2 machine (
p2.xlarge
,ubuntu 16.04
) on which I've installed cuda, python3, numpy andlibKMCUDA
with pip.Here's how I installed CUDA:
And here's
test.py
: