前面换了电脑重新配电脑的环境,各种版本问题,快把我搞崩溃了,总结一下整个过程,需要的小伙伴可以看一下。
一般配置这些环境的顺序是:先电脑安装合适的显卡驱动,然后安装cuda、然后安装cudnn、然后安装对于版本的pytorch。但是这样有一个问题,就是pytorch的版本往往跟不上cuda的更新,这就会导致配置pytorch 和 torchvision的时候各种版本不兼容,各种问题。比如这个文章发布的时候,pytorch支持最高cuda12.1,而此时我的显卡RTX3050,可以安装cuda12.4,如果我先安装cuda的话,就可能会有各种问题。经过无数的折磨之后,我总结一个正确的套路
第一步:电脑安装合适的显卡驱动,在官方驱动 | NVIDIA这里,根据你的显卡型号,下载正确的驱动并完成安装;
第二步:按下win+r,输入cmd,打开命令行在里面输入 nvidia-smi ,然后就会显示你当前最大支持的cuda版本。我的输入之后显示这样,最大的cuda版本就是12.4。
第三步:在pytorch官网