1、遇到的问题:
Failed to build cupy
ERROR: Could not build wheels for cupy, which is required to install pyproject.toml-based projects
安装cupy的过程中一直报错,尝试了pip和conda的方法都没有解决。在百度查看了各种教程也没有很好的方法,探索了一番终于解决了这个问题。
2、解决办法
首先输入:nvidia-smi
查看CUDA的版本
查找到的CUDA版本:12.2.
然后进入cupy的官方主页:https://cupy.dev/
# For CUDA 11.2 ~ 11.x pip install cupy-cuda11x# For CUDA 12.x pip install cupy-cuda12x# For AMD ROCm 4.3 pip install cupy-rocm-4-3# For AMD ROCm 5.0 pip install cupy-rocm-5-0
最终成功安装。