!!!如果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的正确安装安装