请先安装好Anaconda,Python版本为3.6.1。
接着安装好numba
包。
conda install numba
然后,我们需要下载CUDA Toolkit,下载地址:https://developer.nvidia.com/cuda-75-downloads-archive
我用的Mac,所以选择的Mac版本。下载完后,根据提示进行安装。(安装路径,Mac的话,应该在/Developer/NVIDIA/CUDA-7.5/
路径下)
Test
新建文件 test.py
,写入代码:
"""
Python CUDA编程
"""
from __future__ import print_function, division, absolute_import
from timeit import default_timer