安装:
安装之前要查看自己已经安装CUDA的版本
在cmd下运行:
nvcc -V
依据自己版本修改。例如:
然后再开始安装cupy:
显示release 9.2 --》
pip install cupy-cuda92
如果 release 10.2--》
pip install cupy-cuda100
问题:
1.输入安装命令后出现ERROR: No matching distribution found for cupy-cuda92
解决办法:添加镜像
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple cupy-cuda92