終於成功配置numba cuda

很多年前就關注了numba，numba的gpu加速以前叫numba pro，是收費的，後來整合進了numba。但是很遺憾，我從來沒有成功配置過numba的cuda。終於在今天，完成了這一多年來一直失敗的配置過程。

廢話少說，配置cuda主要有以下幾點：

安裝cuda toolkit我比較喜歡新的，裝的是9.1版本。安裝完成後安裝程式會自動配置好環境變數。

設定numba所需的環境變數

環境變數設定好後看看numbapro_libdevice目錄下是否有名為libdevice.compute_xx.10.bc (xx=20,30,35,50)的檔案，如果沒有可以到matlab的bin目錄下去尋找。matlab裡面啥都有。。啥都有。。啥都有。。

環境設定好以後就可以試試numba的example了。numba的example很簡單，mandelbrot集，鏈結如下：

其中使用cpu的numba.autojit無需配置，直接就能達到10倍以上的加速效果，而使用gpu的numba.cuda.jit(device=true)可以加速至上千倍。

在只安裝了cuda toolkit的情況下直接執行上面的example會出現以下錯誤：

oserror traceback (most recent call last) d:\winpython-64bit-3.6.3.0qt5\python-3.6.3.amd64\lib \site-packages\numba \cuda \cudadrv \nvvm.py in __new__(cls) 110 try: --> 111 inst.driver = open_cudalib('nvvm', ccc=true) 112 except oserror as e: d:\winpython-64bit-3.6.3.0qt5\python-3.6.3.amd64\lib \site-packages\numba \cuda \cudadrv \libs.py in open_cudalib(lib, ccc) 40 if path is none: ---> 41 raise oserror('library %s not found' % lib) 42 if ccc:

oserror: library nvvm not found

順著提示找到site-packages\numba\cuda\cudadrv\libs.py，發現其中有個test()函式。其中最重要的就2個函式，get_cudalib get_libdevice，定義如下：

def
get_cudalib
(lib, platform=none):
if lib == 'nvvm'
and os.environ.get('numbapro_nvvm'):
return os.environ.get('numbapro_nvvm')
libdir = os.environ.get('numbapro_cudalib')
candidates = find_lib(lib, libdir, platform)
return max(candidates) if candidates else
none
defget_libdevice
(arch):
libdir = (os.environ.get('numbapro_libdevice') or
os.environ.get('numbapro_cudalib'))
pat = r'libdevice\.%s(\.\d+)*\.bc$' % arch
candidates = find_file(re.compile(pat), libdir)
return max(candidates) if candidates else
none

其中一共出現了3個環境變數，即前述的3個待新增的環境變數。新增好numbapro_cudalib後，執行get_cudalib('cublas')有：』d:\cuda\v9.1\bin\cublas64_91.dll』，則numbapro_nvvm應該也指向dll檔案，即'd:\\cuda\\v9.1\\nvvm\\bin\\nvvm64_32_0.dll'。

觀察get_libdevice的定義和test()函式中的archs = 'compute_20', 'compute_30', 'compute_35', 'compute_50'可知，其需要名為'libdevice.compute_xx.d+.bc'的檔案。

我在裝好cuda9.1後發現libdevice資料夾裡面並沒有這些檔案，於是用listary搜尋了一下，發現matlab裡面就有！萬能的matlab\bin裡面的dll太全了！啥都有！啥都有！啥都有！

配置完成後，enjoy!

終於成功配置numba cuda

終於買到了飛向成功這本書

Matlab TLD終於執行成功了，淚奔。。

nutch配置成功

終於成功配置numba cuda

終於買到了 飛向成功 這本書

Matlab TLD終於執行成功了，淚奔。。

nutch配置成功

相關推薦

終於買到了飛向成功這本書