Cuda: bibliothèque nvvm introuvable

J’essaie d’exécuter le code ci-dessous mais une erreur est signalée:

NvvmSupportError: libNVVM est introuvable. Ne conda install cudatoolkit : bibliothèque nvvm introuvable

Mon environnement de développement est: Ubuntu 17.04, Spyder / Python3.5 et j’ai installé via conda (numba et cudatoolkit). GPU Nvidia (GTX 1070 et GTX 1060).

 import numpy as np from timeit import default_timer as timer from numba import vectorize @vectorize(["float32(float32, float32)"], target='cuda') def VecADD(a,b): return a+bn = 32000000 a = np.ones (n, dtype=np.float32) b = np.ones (n, dtype=np.float32) c = np.zeros(n, dtype=np.float32) start = timer() C = VecADD(a,b) print (timer() - start) 

Est-ce que quelqu’un sait comment résoudre ce problème?

Ce qui a fonctionné pour moi exactement sous le même scénario était d’inclure les éléments suivants dans le .bashrc (j’utilise actuellement cuda-9.0). Ne vous laissez pas rebuter par le NUMBAPRO dans le nom de la variable – cela fonctionne pour numba (du moins pour moi):

 export NUMBAPRO_NVVM=/usr/local/cuda-9.0/nvvm/lib64/libnvvm.so export NUMBAPRO_LIBDEVICE=/usr/local/cuda-9.0/nvvm/libdevice/