有效解决torch.cuda.is_available()为false问题

一、背景

 

第一次根据pytorch官网给出的命令，结合清华镜像源安装：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装过程挺快，看到了torch、torchvision、torchaudio的安装过程，但从始至终都没看到含有cuda字眼的安装代码，心生疑惑，果不其然：

二、解决方案

至于怎么找出这个原因的不再赘述，直接上结论：清华镜像源中并没有cu116这个版本（python3.6环境）。

清华大学开源软件镜像站

发现有cuda11.1和cuda11.3，于是卸载之前安装的版本：

pip3 uninstall torch torchvision torchaudio

将安装代码中的116修改为113：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 -i https://pypi.tuna.tsinghua.edu.cn/simple

此时，安装过程中出现了cu113，预感要成功了。

 果然，问题成功解决。

三、备注

不一定所有torch.cuda.is_available()为false都是这个原因，这是作为新手的我第一次安装pytorch遇到的问题，希望能够帮助到广大铁子！