Support CUDA 10.1? (No such CUDA device exists)

src-d / kmcuda

Large scale K-means and K-nn implementation on NVIDIA GPU / CUDA

Other

783 stars 144 forks source link

Support CUDA 10.1? (No such CUDA device exists) #85

Closed ilgrad closed 5 years ago

ilgrad commented 5 years ago

Hi, after installing kmcuda on python3.7, test.py don't work. cuda version: 10.1 device: Quadro RTX 5000 build flags: -DCMAKE_BUILD_TYPE=Release -DCUDA_ARCH=75 python version: 3.7

######### ValueError: No such CUDA device exists FAILED (errors=40) pute capability mismatch for device 0: wanted 6.1, have 7.5 you may want to build kmcuda with -DCUDA_ARCH=75 (refer to "Building" in README.md) reassignments threshold: 130 #########

ilgrad commented 5 years ago

all work, need also add path in .bashrc ########### export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64" export CUDA_HOME=/usr/local/cuda export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ##########