m0_69358860的博客

原创 将GPU版本的torch和torchvision下载至本地进行安装，解决torch.cuda.is_available()为False的情况，本文主要解决CPU版本的torch问题。

这是困扰我很久的问题，今天花了一点时间去研究一下为什么我的cuda和cudann安装完成了，却不能使用GPU，原因是我的torch和torchvision是cpu版本的，必须下载GPU版本的才能使torch.cuda.is_available()为True,本文使用将文件下载到文件夹后进行安装。此时对应是12.1，之前我是11.1，不知道怎么更新到12.1了， 所以我还是采用11.1版本，此时您需要根据你的版本，选择手动下载GPU版本的torch和torchvision。下图是正确安装后的效果。