应该安装哪个版本的pytorch

img


第一次搞深度学习,下载的pytorch调用不了我的gpu,版本错了,但是好像没有cuda12.0的版本啊😭

你的显卡是比较老还是比较新,一般来说,安装同时代的 pytorch (cudnn)总没错。

你是啥显卡,你这里的是你显卡硬件驱动支持的cuda版本最高到12.x,只要cuda版本不高于这个版本就可以正确运行cuda,而不是你的cuda版本是12.0.一般来说,cuda11.x的版本之间可以兼容,所以你安装cuda11.8(目前最新11.x)就行(其他版本也可以,但是最好要比你的pytorch要求的cuda版本高),现阶段不建议直接上cuda12,太新的话兼容性并不好。
另外,调用不了还可能是你的cuda或者cudnn错误

img


我也是rtx3060,安装的这个版本,供参考,有用的话,请采纳,谢谢