本文档详细记录了在安装RTX3050显卡时遇到的CUDA、cudnn和pytorch版本兼容问题及解决过程。首先从官方下载并安装显卡驱动，然后通过命令行确认支持的最大CUDA版本。接着，下载对应的CUDA工具包和cudnn，并将其文件夹放入CUDA安装目录。最后，在Python环境中安装与CUDA版本兼容的pytorch。注意，安装pytorch前需确保Python版本不低于3.8。
前面换了电脑重新配电脑的环境，各种版本问题，快把我搞崩溃了，总结一下整个过程，需要的小伙伴可以看一下。
一般配置这些环境的顺序是：先电脑安装合适的显卡驱动，然后安装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官网
最低0.47元/天 解锁文章
1098
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
