CuPy 11.2.0for CUDA 11.1(Python 3.9,Windows 64位)
CUDA 加速的cupy
版本,用起来是真香,是你手头有 NVIDIA 显卡时,速度飙升不是吹的。cupy_cuda111-11.2.0-cp39-win_amd64.whl
这个轮子是专门给 Python 3.9 和 CUDA 11.1 配的,兼容性还挺稳。如果你用numpy
做科学计算,那cupy
就是上 GPU 的捷径,语法几乎一模一样,迁移起来毫不费劲。
用cupy
配合tensorflow
或者scipy
效果也不错,大型矩阵分分钟的事,响应也快。要注意一点:CUDA 版本得对上,不然装完直接跑不起来,白折腾一圈。
如果你平时用numpy
、pandas
、matplotlib
画图或者搞点科学可视化,可以一起装好下面这些库,配合使用更顺手:
如果你刚好在搞深度学习项目,tensorflow 2.14.0也可以顺带装上,兼容度还行,能直接用到cupy
的底层加速。
哦对了,.whl
文件别直接双击,记得用pip install
安装:
pip install cupy_cuda111-11.2.0-cp39-win_amd64.whl
装之前先确认一下你有 CUDA 11.1 环境,避免装了跑不了。
78.41MB
文件大小:
评论区