linux Pytorch（GPU版本）后出现torch.cuda.is_available()返回False，解决方案来啦

一级目录

二级目录

第一次的下载过程

我当时下载GPU版本的pytorch的时候，就是直接从官网首页给的命令下载的，如下图：但是好像他们给出的命令默认安装的是CPU版本。

如何判断是否是CPU的版本

source activate 虚拟环境的名称
conda list

这样显示该虚拟环境所安装的包：


这些都表示了下载的pytorch是CPU版本而不是GPU版本的，如果是GPU版本的话后面跟的应该是cuxxx

解决方案

既然是官网首页给出的命令有问题，那就从其他地方找命令：旧版本。从这里面可以找到以前版本的命令，因为我的主机里已经安装了cuda，并且最高可运行的版本是12.0，查看电脑是否安装了cuda：

nvidia-smi

查看cuda版本：

nvcc --version

这些都检查之后，去那个网页找到想要下载的cuda版本，比如我想下载11.7的cuda版本：

我把这条命令复制到命令行中，回车，等待安装。
安装成功之后，检验：

输出True，表示安装成功。