一、查询自己电脑上的cuda版本方法:
1.在windows的cmd里查询显卡cuda的版本号,命令行输入:nvidia-smi
2.在NVIDIA控制面板上寻找自己电脑上cuda的版本
二、安装支持的cuda的python cupy库
因为我的电脑上为cuda11.4,所以使用cuda114,不同的版本需要修改这个值。但是显示没有这个版本。
后面尝试输入 pip install cupy-cuda11x 这个命令即完成安装 。即
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple cupy-cuda11x