一、解决方法
目前主要的解决方法:
-
确定显卡是否支持CUDA,https://www.geforce.com/hardware/technology/cuda/supported-gpus。(本人电脑显卡为RTX2050,应该是支持的)
-
确保驱动升级到最新,一般高于396.26就可以。
-
由于清华镜像问题,下载的pytorch为cpu版本。
二、问题分析
在通过显卡和驱动检查都没有问题的时候,在pytorch环境中输入:
conda list
检查当前环境的安装包:
![](https://img-blog.csdnimg.cn/img_convert/bcd597a7fb40d2bae4630d18d60693f6.png)
可以看到此时的pytorch包为cpu版本。
(原因应该是在安装时,由于solving environment一直失败,在解决该问题时使用了清华镜像,而不是在官网获取的命令)
三、具体解决方法
可以使用以下命令先进行卸载:
conda uninstall pytorch
在根据官网提供的命令安装对应的版本。我的电脑cuda11.6选择了低一点的版本cuda11.3
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
pytorch官网:Previous PyTorch Versions | PyTorch
这时输入:
torch.cuda.is_available()
返回true
![](https://img-blog.csdnimg.cn/img_convert/0b8207504d9485acfa333e01217713f3.png)
pycharm也返回true:
![](https://img-blog.csdnimg.cn/img_convert/9fac1769142ec0bce7ce2c11bfd849e5.png)
其他解决方法参考:
1.有关于cuDNN下载: torch.cuda.is_available()返回false——解决办法_cuda available false-CSDN博客
2.torch.cuda.is_available函数总返回False_torch.cuda.is available返回false-CSDN博客