作者:RayChiu_Labloy
版权声明:著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处
之前用tensorflow的时候,对显卡、python版本、cuda版本之类的限制比较死,现在用pytorch,依然很小心cuda之类版本的对应关系,但是最近在低配置环境测试的时候发现有时候cuda版本不够用,太低了,我目前在GTX1050显卡上测试,之前和大家一样最早用cuda9.0 ,后来小心意义的换成cuda10.0,一段时间后发现还是不好用,这次我仔细看了NVIDIA官网介绍Release Notes :: CUDA Toolkit Documentation,发现显卡只和cuda驱动有关:
找找我的GTX1050能装的驱动 :
现在最新的驱动是496,理论上可以装很高的cuda版本,但是我觉得cuda10.2应该够用,因此下载cuda10.2:CUDA Toolkit 10.2 Download | NVIDIA Developer
安装即可。
【如果对您有帮助,交个朋友给个一键三连吧,您的肯定是我博客高质量维护的动力!!!】