利用torch安装CuPy,使得numpy(np)在CUDA(gpu)上加速。
前言标题名字是很奇怪,没关系,重点看内容。正常安装CupPy需要较为复杂的cuda环境配置,可以参考文章——UDA环境配置。如果你觉得去官网安装CUDA Toolkit太麻烦,那我们可以利用pyotch的cudatookit来保证Cupy的正常运行。正文CuPy官网 官网给出了详细的安装操作,但是需要手动安装CUDA Toolkit,如果你没有实践过,这也许会比较难。官网给出了相应版本对应的安装命令:我电脑是11.1,所以这里执行下面命令即可pip install cupy-cuda
原创
2022-05-11 14:18:57 ·
8983 阅读 ·
4 评论