Tvm.gpu().exist returen False


#1

Hi,

I’m having problem with linking my GPU to tvm.

Python 3.5.2 (default, Nov 12 2018, 13:43:14) 
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tvm
>>> tvm.gpu().exist
False

but it seems that Cuda and Nvidia driver is installs correctly and functioning:

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105

and :

$ nvidia-smi 
Fri Aug 23 15:23:38 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.130                Driver Version: 384.130                   |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 1080    Off  | 00000000:01:00.0  On |                  N/A |
| 27%   36C    P8     8W / 180W |    609MiB /  8112MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      1409      G   /usr/lib/xorg/Xorg                           350MiB |
|    0      3106      G   compiz                                       153MiB |
|    0      3707      G   ...uest-channel-token=17673461797500912344   102MiB |
+-----------------------------------------------------------------------------+

also I am able to build and run cuda test programs usin nvcc

anyone ?

Thanks,
Dror


#2

Did you build TVM with CUDA enabled?