最后查了一下发现我torch安装在conda虚拟环境下了,而项目用的是终端的环境,重新下载配置就好了
这是我电脑支持的CUDA版本
这是我安装的CUDA版本
这是我安装的Pytorch 版本
然后我安装的cuDNN也是支持CUDA12.X的
但是在运行torch.cuda.is_availavle()时返回值是False,并且我训练模型时,用到的是CPU,我的GPU一直处于空闲状态(电脑有独显)
最后查了一下发现我torch安装在conda虚拟环境下了,而项目用的是终端的环境,重新下载配置就好了
这是我电脑支持的CUDA版本
这是我安装的CUDA版本
这是我安装的Pytorch 版本
然后我安装的cuDNN也是支持CUDA12.X的
但是在运行torch.cuda.is_availavle()时返回值是False,并且我训练模型时,用到的是CPU,我的GPU一直处于空闲状态(电脑有独显)