我是从torch下手的,因为我确认torch,cuda和cudnn的版本都对应无误。如果确认这点,可以看看下面这点是否有帮助:
如果直接用pip install torch,则会默认安装cpu版本的。应该从官网找到gpu版本的下载
另外可以看看这个比较全面的教程:深度学习第一步——Pytorch-Gpu环境配置:Win11/Win10+Cuda10.2+cuDNN8.5.0+Pytorch1.8.0(步步巨细,少走十年弯路) - 古月居
我是从torch下手的,因为我确认torch,cuda和cudnn的版本都对应无误。如果确认这点,可以看看下面这点是否有帮助:
如果直接用pip install torch,则会默认安装cpu版本的。应该从官网找到gpu版本的下载
另外可以看看这个比较全面的教程:深度学习第一步——Pytorch-Gpu环境配置:Win11/Win10+Cuda10.2+cuDNN8.5.0+Pytorch1.8.0(步步巨细,少走十年弯路) - 古月居