
numpy-gpu:利用Copperhead在NVIDIA GPU上运行numpy
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
numpy-gpu项目旨在通过Copperhead库将numpy代码迁移至NVIDIA GPU上执行,显著提升大规模数值计算任务性能。
在 NVIDIA GPU 上使用 numpy(通过 Copperhead)。部署:CUDNN 4.8 CUDA 6.5
为了使它工作,我必须:
将铜头源代码中 move() 函数的所有用法更改为 std::move() 以避免与 boost::move() 混淆;
在 cuda 或 thrust 包含文件中删除对 GCC 版本的限制;
在 Copperhead 源码中的某处添加对 Thrust 重新标记的包含。
安装:
首先,安装 CUDA 6.5,然后:
(使用 virtualenv 简而言之;源 nutshellbinactivate)
pip 安装 numpy codepy
cd copperhead
python setup.py build
python setup.py install
或者如果您信任它:source setup.sh
用 GPU 测试 numpy:
cd copperhead 样品
全部评论 (0)
还没有任何评论哟~


