！避雷：print(torch.cuda.is_available())返回False
Posted 闭关の阿洁
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了！避雷：print(torch.cuda.is_available())返回False相关的知识，希望对你有一定的参考价值。
！！！如果torch.cuda.is_available()返回false
如果你遇到这个问题，不用担心你的步骤是不是错了，因为你被conda镜像安装给坑了。你以为下载的是GPU版本，其实镜像下载的是cpu版本，你必须手动去官网用pip下载安装。
1、先查看cuda\\cudnn版本是否都对。
执行nvcc --version
是cuda11.6版本。
2、查看该虚拟环境下安装的pytorch\\torchvision版本是否对应cuda11.6。
显然，出了问题。
能看到，安装的torch是cpu版本。
如果你的CUDA，cuDNN版本都对，只有Pytorch安装成了CPU的，那么先对它进行卸载，然后再安装。
卸载方法：删除该虚拟环境的lib->site-packages中的torch、torchaudio和torchvision的六个文件，
再重新安装正确的torch及torch vision。
安装的步骤见上一篇：二、安装cuda与pytorch的正确安装安装
以上是关于！避雷：print(torch.cuda.is_available())返回False的主要内容，如果未能解决你的问题，请参考以下文章