本文主要是介绍Python下载cuda包失败后到成功(方便使用GPU加速运算,显著提高代码运行速度),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
一、查询自己电脑上的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
这篇关于Python下载cuda包失败后到成功(方便使用GPU加速运算,显著提高代码运行速度)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!