亲测有效解决torch.cuda.is

2023-11-06 03:28| 来源: 网络整理| 查看: 265

文章目录解决torch.cuda.is_available()返回False出现返回False的原因问题1：版本不匹配问题2：错下成了cpu版本的（小编正是这种问题）解决方案方案一方案二

解决torch.cuda.is_available()返回False 出现返回False的原因问题1：版本不匹配

电脑实际安装的CUDA版本大于已安装的pytorch所对应的CUDA的版本注意：小于等于是可以的，但是不能大于

问题2：错下成了cpu版本的（小编正是这种问题）

可能使用了清华源镜像，但是清华源镜像中所对应的pytorch没有cuda版本的，这就造成了下载成了cpu版本的错误。

解决方案方案一

针对于问题一，降低pytorch的版本，一般不建议去修改CUDA的版本，因为CUDA的版本会极有可能影响其他部分的使用。

方案二

针对于问题二，

打开 Anaconda Prompt ，输入conda list，如下图所示

在这里插入图片描述 2.卸载pytorch，输入conda uninstall pytorch

3.重新安装pytorch

官网链接：https://pytorch.org/ 在这里插入图片描述

在Anaconda Prompt，直接输入以下命令，注意不要用清华源（下面命令加了-c pytorch，就意味着不再是使用清华源）

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

4.检验pytorch是否可用

在这里插入图片描述

【本文地址】

今日新闻