五步解决torch.cuda.is_available()返回False的问题

最新推荐文章于 2025-04-17 11:40:55 发布

Flora_Olivia

最新推荐文章于 2025-04-17 11:40:55 发布

阅读量2.8w

点赞数 34

分类专栏：怡宝升级记文章标签： cuda gpu pip 深度学习 pytorch

本文链接：https://blog.csdn.net/Flora_Olivia/article/details/104486548

版权

当`torch.cuda.is_available()`返回False时，问题可能出在PyTorch版本与CUDA版本不匹配。通过检查PyTorch实际安装版本，发现与预期不符。通过卸载旧版本，手动下载对应CUDA版本的安装包并安装，成功解决了问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

First of all，感谢博主：dysljsjsq 链接

基于上次发的帖子，我是用Anaconda创建虚拟环境装的pytorch。

可以看到我在官网上选择的是Stable(1.4)，cuda10.1，linux，conda安装

conda install pytorch torchvision cudatoolkit=10.1 （我把-c pytorch去掉了，这样的方式安装快）
pytorch官网版本选择然后，我用了一段时间，import torch完全没问题，可以创建tensor类型，完全ok。但我一用到和cuda有关的语句，就报错，说不存在cuda。我一开始不懂为什么，查了很久之后，也尝试了很多博主说的驱动版本可能不对，更新版本之类的，重装过Nvidia。
驱动版本等信息驱动版本和cuda版本完全没问题。顺便附上版本对应表