python cupy库的安装

  • 参考教程
  • 问题
  • 解决
  • 注意

参考教程

https://www.jianshu.com/p/dc8d13cb0c80
其实不止这一篇，总结就是3种方法1.pip2.whl3.conda，最方便的无疑是conda。

问题

  1. pip安装：依照官网https://docs.cupy.dev/en/latest/install.html，我的cuba版本是11.8，所以选择pip install cupy-cuda11x安装完成，但应该是欠缺一些依赖包，没有运行成功。
  2. whl安装：即下载whl轮子，有点麻烦，没有尝试。
  3. conda安装：直接conda install cupy也不行，应该是版本不符合，于是尝试conda install cupy-cuda11x，也不行。

解决

按照教程conda install -c conda-forge cupy，即使没有限制版本，但成功了。

注意

库是否能安装成功要看，1.环境的python版本2.环境中其他库版本，与所要安装的库版本是否兼容。因为conda默认会安装较新版本，所以我建议遇到库实在安装不上情况，创建较新版本anaconda环境，python选最新版本，再按照conda install -c conda-forge xxx 依次安装库。