torch.cuda.is_available()输出false_torch.cuda.is available()显示false.

作者：正经夜光杯 | 2024-08-13 01:51:06

踩

torch.cuda.is available()显示false.

问题描述

模型训练时无法使用gpu
检查发现torch.cuda.is_available()输出为false

原因分析：

powershell里打开对应的虚拟环境发现torch版本又给自动装成cpu版本了
在这里插入图片描述

解决方案：

右键打开Nvidia 控制面板，工具栏中点击帮助，找到系统信息(System Information)，显示cuda现在安装版本是12.2：
在这里插入图片描述
把当前虚拟环境下的cpu版本的pytorch卸载
pip uninstall torch

去 pytorch官网下载cuda对应的pytorch（在这之前要下载好cuda、cudnn等，具体如何对应下载可以自行搜索）

把给出的command运行就开始下载喽

安装完成之后再看一下torch.cuda.is_available()输出就是true啦
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/正经夜光杯/article/detail/972585